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 182.
This should be the command you are looking for:
conda install pytorch torchvision cuda90 -c pytorch