I am facing same issue but when I am trying to update pytorch to 0.2.0 its updated but when I checked in code its showing that its using pytorch 0.1.12
issue:
pytorch_issue.png1858×1056 198 KB

my pytorch-gpu version 0.1.12
pytorch version 0.2.0
is it using pytorch-gpu? if so I uninstalled pytorch-gpu then its giving me that no module torch.
how can i fix this issue.