1.下 cuda toolkit 8.0
CUDA Toolkit 8.0 GA1 Download | NVIDIA Developer
2.下 cuDNN 6.0
cuDNN Archive | NVIDIA Developer
3.
把 cuDNN 里的 bin、include、lib 里的东西放到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0 相应文件夹里
4.
5.
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
6.测试
import torch
print(torch.cuda.is_available())