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
NVIDIA GeForce GT 730 配置 CUDA

3.

把 cuDNN 里的 bin、include、lib 里的东西放到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 相应文件夹里

4.

NVIDIA GeForce GT 730 配置 CUDA

5.

conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch

6.测试

import torch
print(torch.cuda.is_available())
上一篇:图像分割之Unet网络详解


下一篇:tensorflow2.0 tf.data 简单示例 以及实现mnist手写数字识别