环境配置: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

验证

版本验证

  • 在命令行中输入
nvcc -V

会观察到如下版本信息:

figure.7

PyCharm验证能够使用CUDA

import torch

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

StriveZs

用爱发电

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值