最近重新配置了一台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
这样终于可以成功运行