1.下 cuda toolkit 8.0
https://developer.nvidia.com/cuda-80-download-archive
2.下 cuDNN 6.0
https://developer.nvidia.com/rdp/cudnn-archive
3.
把 cuDNN 里的 bin、include、lib 里的东西放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 相应文件夹里
4.
5.
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
6.测试
import torch
print(torch.cuda.is_available())