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

本人深度学习框架小白一枚，在安装使用深度学习框架过程中遇到了不少问题，幸好有师兄的帮助，才得以成功安装。同时也深感对于新入门的小伙伴，在安装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也是支持的） 这样我们需要安装的软件包已经下载完毕了。