win10 英伟达显卡MAX150 安装GPU版pytorch（一）选择合适的CUDA、CUDNN和torch

最新推荐文章于 2022-08-22 12:46:07 发布

312Benson

最新推荐文章于 2022-08-22 12:46:07 发布

阅读量4.3k

点赞数 6

分类专栏：深度学习框架安装文章标签： pytorch NVIDIA显卡MX150

本文链接：https://blog.csdn.net/weixin_44001860/article/details/89552541

版权

深度学习框架安装专栏收录该内容

1 篇文章 0 订阅

订阅专栏

本人深度学习框架小白一枚，在安装使用深度学习框架过程中遇到了不少问题，幸好有师兄的帮助，才得以成功安装。同时也深感对于新入门的小伙伴，在安装pytorch过程中的不易，特此将经验分享一下，如果觉得不错，请点个赞吧。
本人电脑的基本配置是：2018年4月份买的电脑，英伟达显卡MAX150，固态硬盘 256G，Intel i5-8250u, 使用的是python3.7 ，IDLE是pycharm
安装的pytorch 的版本是：torch-1.0.1-cp37-cp37m-win_amd64
安装的内容：CUDA CUDNN 和torch
具体的安装步骤如下：
1、你的电脑必须有英伟达(NVIDIA)的显卡,这是必备的基本条件。第一步，有的博客说你需要到nvidia的官网查看你的显卡是否支持的CUDA，个人认为是没有必要的。你需要做的是在桌面右键打开控制面板，在这里插入图片描述
点击左下角的“系统信息”→再点击“组件”
这样就查到了你的显卡是否支持CUDA以及适合的版本是哪种。之后就可以去NVIDIA的官网https://developer.nvidia.com/cuda-toolkit-archive下载对应版本。（这是我的显卡支持的版本，但是我不建议使用太新的版本。所以我下载的是9.2版的。）在这里插入图片描述
其次就是下载对应版本的CUDNNhttps://developer.nvidia.com/cudnn。注意，两者的版本一定要对应！在下载CUDNN时需要先进行注册，比较简单一分钟就完事了。之后就下载相应的版本就行了。

然后到pytorch的官网https://pytorch.org/下载对应版本的torch 在这里插入图片描述
这里需要注意选择你的电脑系统，安装包类型、你使用的编译语言以及CUDA的版本。（虽然我下载的CUDA是9.2的，但是这里选择9.0的torch也是支持的）
这样我们需要安装的软件包已经下载完毕了。