https://pytorch.org/get-started/locally/
本人环境是
CUDA 9.0
To install PyTorch via pip, and you are using CUDA 9.0, use the following command, depending on your Python version:
Python 3.6
pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
pip install torchvision