windows下配置pytorch

参考:

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,但是调用成功。

完成。

原文地址:https://www.cnblogs.com/Oscar67/p/11164448.html