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

最新推荐文章于 2024-12-13 08:42:12 发布

312Benson

最新推荐文章于 2024-12-13 08:42:12 发布

阅读量4.4k

点赞数 6

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/weixin_44001860/article/details/89552541

本人深度学习框架小白一枚，在安装使用深度学习框架过程中遇到了不少问题，幸好有师兄的帮助，才得以成功安装。同时也深感对于新入门的小伙伴，在安装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时需要先进行注册，比较简单一分钟就完事了。之后就下载相应的版本就行了。