史上最详细最全pycharm使用GPU训练模型总结(Pytorch版)

一、硬件配置：

1、查看驱动：

打开命令行win+r，输入cmd，在命令行输入：nvidia-smi 查看信息

第二个方框显示电脑具备独立显卡，并显示型号为：NVIDIA GeForce GTX 1050 Ti，如果这里没有显示显卡信息说明本机没有独立显卡，这可能需要换电脑装显卡或者只能使用GPU进行训练。
第一个方框显示我的驱动是536.67，根据下图可以看到驱动只要大于451.22就支持CUDA11，,pytorch最新本已经不支持CUDA10,如果驱动版本低于451,可以升级驱动，点击此处下载驱动，下面是CUDA和显卡驱动对应的版本：

也可以参考官方文档：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues

2、CUDA安装：

NVIDIA官网下载:https://developer.nvidia.com/cuda-toolkit-archive
因为国外网站的原因，官网会一直刷新失败，此时使用一个小方法，将网址后缀做些更改，就解决了显示和下载的问题。
将官网地址https://developer.nvidia.com/cuda-toolkit-archive里面的.com改成.cn
选择合适自己的版本：

选择好电脑系统版本信息后开始下载：

3、cuDNN安装：

点击此处，自行使用邮箱注册账号或者登录账号

在首页点击下载，找到CUDA-X Libraries

下滑找到深度学习库：