使用 NVIDIA GeForce 940MX GPU环境安装

笔记本的GPU型号 940MX，GPU资源要利用起来。

匹配的CUDA版本为8.0（下载地址：CUDA Toolkit Archive | NVIDIA Developer），对应的cudnn版本为5.1（下载地址：cuDNN Archive | NVIDIA Developer），cuda正常安装即可，安装完成后可检查环境变量中的是否有nvidia相关路径。

可在cmd中使用 nvcc -V 测试是否安装成功。

cudnn的包解压后，将对应文件夹中的文件复制到cuda对应路径的文件夹中，cuda安装位置：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

下载python 3.6：Index of /ftp/python/

下载对应的pytorch 1.0.0 whl，地址：https://download.pytorch.org/whl/cu80/torch_stable.html

同时需要安装torchvision：pip install torchvision==0.2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

nvidia-smi命令问题：

重新安装cuda后，nvcc -V 命令可用，执行nvidia-smi时会提示：Failed to initialize NVML: N/A

检查后发现是C:\Program Files\NVIDIA Corporation\NVSMI文件夹中没有文件，从网上下载了NVSMI.zip包，将内容解压到NVSMI文件夹下即可。（下载地址：百度网盘 请输入提取码，提取码：wy6l，感谢资源上传者）。

参考：

Windows+NVIDIA GeForce 940MX+Pytorch搭建深度学习运行环境 - 简书

https://zhuanlan.zhihu.com/p/627204574

pytorch 1.0.0 wheel 安装-CSDN博客

Windows NVIDIA Corporation下没有NVSMI文件夹解决方法-CSDN博客