win10+英伟达mx150安装cuda,cudnn,GPU版本pytorch（荣耀magicbook2018）

好好学习多做笔记

已于 2022-12-22 10:10:29 修改

阅读量2.8k

点赞数 2

文章标签： pytorch 深度学习 python

于 2022-08-22 12:46:07 首次发布

本文链接：https://blog.csdn.net/qq_42675237/article/details/126463574

版权

最近打算给笔记本装一个gpu版本的pytorch,装了有半天才搞完，中间也出现了不少问题看了不少博客，现在总结一下出现的问题以及解决办法，希望能给遇到同样问题的提供一些帮助。本人水平有限，有问题还请指正。

我这里装的是pytorch1.6.2版本，CUDA对应9.2，建议各位装pytorch之前，首先确定你确实需要什么版本的pytorch，在根据你设备的具体情况酌情安装。

安装cuda

如果想使用gpu版本的pytorch,首先需要安装一个自己显卡支持的cuda版本
我的电脑显卡是mx150,cpu是i7-8550U,操作系统是win10，所以我这里选择的是cuda9.2版本https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 可以查看安装cuda所需要的显卡驱动版本，查看电脑的显卡驱动版本，右键选择NVIDIA 控制面板->帮助->系统信息
显卡驱动版本只要大于cuda对应的最低版本就行，如果显卡驱动版本过低，可以去英伟达的官网下载对应的驱动版本，然后升级显卡驱动版本，接下来下载我们所需要的对应版本的cuda，这里是下载网址https://developer.nvidia.com/cuda-toolkit-archive

下载完成之后安装即可。

如果在安装的过程中一直出现visual studio integration失败的提示，关于这个问题也查了许多资料，最简单的办法就是选择自定义安装，在cuda组件下取消勾选visual studio integration，这样就不会再安装此模块。
另外安装时还出现了弹窗提示脚本错误，直接无视即可。

安装成功之后可以测试一下，在命令行中输入nvcc -V
能输出版本信息代表安装成功
第二步安装cudnn （未完）需要先注册才能下载，选择对应cuda的版本，这里选择cuda9.
安装anaconda
（安装软件之前，要确定自己电脑所需的python版本，不可贸然下载最新版本，确定版本的方法即就是看Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror中pytorch的所需版本与CUDA 的版本确定python的版本）由于我的电脑已经安装过anaconda就不再重复此操作，我的是python3.8.8，
这个链接中有python版本对应的anaconda的版本号，koverholt/anaconda-version-map: List of Anaconda/Miniconda versions and their corresponding Python versions (github.com)有了版本号，就可以去官网Anaconda | Anaconda Distribution或者我这边提供的清华大学的开源数据库Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror下载。
安装时注意勾选将anaconda加入到环境变量中

打开Anaconda Promt
通过conda创建一个名为pytorch的虚拟环境
```
conda create -n pytorch python=3.8.8
```
可以使用如下命令查看已经创建好的虚拟环境
```
conda info --envs
```
然后激活pytorch虚拟环境
```
conda activate pytorch
```
运行之后命令行前面的(base)会变成(pytorch)表示已经切换到pytorch环境中
安装pytorch
```
pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
```
Previous PyTorch Versions | PyTorch上面代码在此旧版本的pytorch官网中对应版本下载（上面代码在pycharm终端中打开）