Just FYI,
Install script: conda install pytorch torchvision cuda90 -c pytorch
install torchvision as well. But the transforms.py in it has a bug as I found in running some other python code:
Traceback (most recent call last):
File “main.py”, line 74, in
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
TypeError: init() got an unexpected keyword argument ‘root’
The latest version of torchvision reinstalled through python setup.py install works correctly.