ubuntu下安装cuDNN

正确的安装cuDNN方式,其实跟着官方安装说明进行安装就可以了。

#####################################################################

1. 从https://developer.nvidia.com/cudnn上下载cudnn相应版本的压缩包(需要注册或登录)。
   下载cuda对应版本的cudnn:cuDNN Library for Linux

   解压当前的.tgz格式的软件包

tar -xzvf cudnn-9.0-linux-x64-v7.tgz
   解压后的文件夹名为cuda,文件夹中包含两个文件夹:一个为include,另一个为lib64。
   最好把这个cuda文件夹放在home目录下,方便操作。 

#####################################################################

2. 将解压后的lib64文件夹关联到环境变量中。这一步很重要。

sudo gedit ~/.bashrc

   在最后一行加入:

export LD_LIBRARY_PATH=/home/cuda/lib64:$LD_LIBRARY_PATH
   注意lib64文件夹的路径,

   保存更改的文件后,紧接着:

source .bashrc
   再重启一下终端,该步骤可以成功的配置cuDNN的Lib文件。

#####################################################################

3. 配置cuDNN的最后一步就是将解压后的/home/cuda/include目录下的一些文件拷贝到/usr/local/cuda/include中。

   由于进入了系统路径,因此执行该操作时需要获取管理员权限。

sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
   到此cuDNN的配置就完成了。

#####################################################################
  • 7
    点赞
  • 32
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值