查看cudnn是否安装成功_Window10 Pytorch gpu版安装记录

①安装cuda

②安装cudnn

③安装pytorch

安装cuda:查看gpu支持的cuda版本,右键进入NVIDIA控制面板->帮助->系统信息->组件,这里支持CUDA10.1

d226993eaa4f641f24fafaf9576599b2.png

下载10.1版cuda

CUDA Toolkit Archive​developer.nvidia.com

1bef38afc043840639d1f38e379c6767.png

安装cuda:先安装Visual Studio,我这里装的是VS2019,否则会出现安装Visual Studio Integration报错,然后安装cuda,选择精简安装即可,通过在cmd中键入命令 nvcc -V测试是否安装成功。

c8bf85ca5fc44f4cdf96e0974e7c2e50.png

114f46cb44148eb812c3b0c358edfab3.png

安装cudnn:下载10.1版cudnn,解压复制到cudn安装目录:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1下,可以看到include文件夹下包含cudnn.h文件

NVIDIA cuDNN​developer.nvidia.com
d177321ebe324ea94f3f7c05dc71eb41.png

ddf231a5cf3e6acd181e0b804a1bbad9.png

6069ba747a8386b91f0b99cfab69aa2d.png

安装Pytorch

PyTorch​pytorch.org
9c9679fc47a3cc68dc26aec582608a62.png

abbb9f7f843d105cfd3499e132cce8f5.png

018e51c211c47a4f75cb9a75c7ed58f9.png

安装成功

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值