Win10+940MX显卡配置PyTorch1.0.0的痛苦经历

最近重新配置了一台2017年的笔记本电脑，花了好几天才配置好，分享一下惨痛经历

1.安装CUDA

电脑搭载的是GeForce 940MX显卡，对应的最高版本是CUDA8.0

因此需要先在NVIDIA官网下载CUDA8.0+对应的cuDNN并进行安装

2.安装Anaconda

由于CUDA版本过于老旧，因此不支持最新版的PyTorch2.0.0

CUDA8.0支持的最高版本是PyTorch1.0.0，而该版本PyTorch需要在Python3.5~3.7之间的版本下运行

因此选择安装2018年版的Anaconda3-4.3.1

3.安装PyTorch+更新Python

PyTorch1.0.0已经无法通过conda install pytorch==1.0.0安装

因此安装方式为Anaconda Prompt下运行

pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-win_amd64.whl​​​​​​

另外在使用时Python3.6.0版本import torch会报错

我的解决方案是升级为Python3.6.1+NumPy1.14.1

这样终于可以成功运行