安装cudnn

本文介绍了cuDNN的安装与配置方法。先给出安装参考链接,解压文件后,解决使用sudo gedit.bashrc报错问题,在.bashrc中添加路径并保存。接着将cudnn.h文件拷贝到指定路径,重置读写权限完成配置。还提供了查看cuda和cudnn版本的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

安装参考:

https://blog.youkuaiyun.com/Lucifer_zzq/article/details/76675239

先解压:

tar -zvxf archive_name.tar.gz

 

cd ~
sudo gedit .bashrc

 使用中间 sudo gedit .bashrc 报错解决:

远程连接服务器或者工作站的时候,通常不能打开一个GUI图形用户,而且在打开过程中会出现诸如:(gedit:46927): Gtk-WARNING **: cannot open display: 或者: Cannot connect to display 

解决办法:

登录时ssh 后加上-Y

ssh -Y username@IP-远程

在弹出的gedit文档编辑器(.bashrc中)中最后一行加入:

   export LD_LIBRARY_PATH=/your/path/to/cuda/lib64:$LD_LIBRARY_PATH

其中/your/path/to/cudnn/lib64是指.tgz解压后的文件所在路径中的lib64文件夹。
保存更改的文件后,紧接着:

   source .bashrc

配置cuDNN的最后一步就是将解压后的cuDNN文件夹(一般该文件名为cuda)中的include文件夹(/your/path/to/cudnn/include)中的cudnn.h文件拷贝到/usr/local/cuda/include中,由于进入了系统路径,因此执行该操作时需要获取管理员权限。

打开终端,进入/your/path/to/cudnn/include。其中/your/path/to/cudnn/include指的是.tgz解压后的文件所在路径中的include文件夹。例如:

   cd /YOUR/PATH/TO/cuda/include
   sudo cp *.h /usr/local/cuda/include/

其中这里的cuda/include对于我自己的安装来说就是/your/path/to/cudnn/include。因为我将cudnn的.tgz压缩包解压到了home的当前用户的路径下,解压后的文件夹名为cuda。

之后,再重置cudnn.h文件的读写权限:

   sudo chmod a+r /usr/local/cuda/include/cudnn.h

至此,cuDNN的配置就全部安装完成了。


查看cuda版本的方法:

cat /usr/local/cuda/version.txt

查看cudnn版本的方法:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值