【工具】cuDNN安装配置

最新推荐文章于 2024-07-20 11:42:43 发布

菜菜2022

最新推荐文章于 2024-07-20 11:42:43 发布

阅读量473

点赞数

分类专栏： Pytorch 工具使用学习笔记文章标签： cuDNN 英伟达

本文链接：https://blog.csdn.net/kz_java/article/details/122068456

版权

41 篇文章 1 订阅

订阅专栏

26 篇文章 1 订阅

订阅专栏

安装cuDNN之前需要安装cuda，关于cuda的安装可以查看：《cuda安装及验证》

下载

下载地址
在这里插入图片描述 在此之前你还需要注册一个英伟达的账号，才能进行下载哦

在这里插入图片描述
解压之后
找到cuda安装位置，将刚刚解压的cuDNN下的三个文件夹：bin、include、lib下的内容分别复制到cuda对应的同名文件夹下面

这里需要特别注意的是lib文件夹需要再往下一层，将x64文件夹下面的内容复制到cuda的lib的x64文件夹下面：

在这里插入图片描述最后添加环境变量
至此，cuDNN就安装配置完成了

from torch.backends import cudnn
#判断是否安装了cuDNN
print(cudnn.is_available())         #返回True则说明已经安装了cuDNN

返回True说明已经成功安装了cuDNN

关注

专栏目录