环境配置:NVIDIA+Anaconda+PyTorch+PyCharm | StriveZs的博客

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

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

步骤

配置英伟达显卡

安装CUDA

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

figure.3

figure.0

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

figure.1

安装Cudnn

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

figure.2

配置PyTorch

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

figure.4

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

figure.5

  • 安装完成

figure.6

验证

版本验证

  • 在命令行中输入
1
nvcc -V

会观察到如下版本信息:

figure.7

PyCharm验证能够使用CUDA

1
2
3
4
5
import torch

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

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

StriveZs wechat
Hobby lead  creation, technology change world.