upgrade pytorch version