环境配置：NVIDIA+Anaconda+PyTorch+PyCharm

¶步骤

¶配置英伟达显卡

¶安装CUDA

• 首先打开自己的英伟达控制面板查看自己的系统信息，如下图所示：我的是CUDA10.2.95因此需要去英伟达官网下载对应版本的CUDA，下载地址：click here

• 进行安装，基本上就是傻瓜式下一步没什么，等待安装完成即可

¶安装Cudnn

• 首先去英伟达官网下载cudnn(这里需要登录下载)。具体下载链接：click here
• 下载完成之后解压得到一个文件夹(蜜汁感动竟然是压缩包不用安装), 将文件夹中的内容放到在C盘路径为**C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2**的对应文件夹中，直接复制进去就好了

¶配置PyTorch

• 这里我使用的是anaconda来进行包的管理，具体anaconda下载地址： click here
• PyTorch安装，切记这里需要安装和cuda对应版本的PyTorch！
• 首先去PyTorch官网找到安装命令，如下图：

• 然后在Anaconda中输入该命令即可：

• 安装完成

¶验证

¶版本验证

• 在命令行中输入

nvcc -V

会观察到如下版本信息：

¶PyCharm验证能够使用CUDA

import torch

print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())

如果输出结果为True，则表示cuda可以使用了。（建议不用最新版的NVIDIA驱动，否则CUDA版本也会升级 PyTorch不一定会有对应版本 没有对应版本不能正常使用。）