Solved by ptrblck in post #2 
              
              
                How did you install PyTorch? 
Did you compile from source or install the pre-built binaries? 
You can find the install command on the website. 
This should be the command you are looking for: 
conda install pytorch torchvision cuda90 -c pytorch