NVIDIA GeForce GT 730 配置 CUDA

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())