参考:
https://www.jianshu.com/p/59b6b23c3d11
1.查看当前NVIDIA显卡对应的驱动版本:
2.安装CUDA9.0(旧版)
https://developer.nvidia.com/cuda-toolkit-archive
新版本直接上官网下载:
https://developer.nvidia.com/cuda-downloads
3.下载cuDNN(对应CUDA版本)
https://developer.nvidia.com/rdp/cudnn-archive
安装方法就是将对应文件夹内文件拷到cuda安装路径下。
4.pytorch安装
由于显卡版本较低,新版pytorch不支持,所以需要安装旧版pytorch。
https://blog.csdn.net/whyymlm/article/details/79696357
下载地址:(0.4版本)
https://pytorch.org/get-started/previous-versions/
然后安装torchvision:
pip install --no-deps torchvision
卸载:
安装:
https://pytorch.org/get-started/locally/
5.最后测试:
python编辑:
pycharm运行结果:
有warning,但是调用成功。
完成。