Updating pytorch versions?

[Atcold (Alfredo Canziani)]

@Kalamaya, I wrote some notes about this here. See if they help making sense.