Ubuntu安装cuDNN

一、官网下载与自己CUDA版本匹配的安装包 cudnn-(版本号).tgz

二、安装cuDNN

tar xzvf cudnn-(版本号).tgz cuda/[Tab]
sudo cp cuda/include/* /usr/local/cuda-1x.x/include/
sudo cp cuda/lib64/* /usr/local/cuda-1x.x/lib64/

三、验证

cd /usr/local/cuda-1x.x/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
输出最后一句有CUDA版本号

  • 4
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,cuDNN是NVIDIA专门为深度学习框架提供的加速库,它可以优化卷积神经网络的训练和推理过程。如果你想在Ubuntu上使用cuDNN,需要安装CUDA和cuDNN两个软件包。 下面是具体的安装步骤: 1. 安装CUDA 首先,你需要先安装CUDA,可以从NVIDIA的官网上下载对应版本的CUDA安装包。下载完成后,打开终端,进入到安装包所在的目录,执行以下命令进行安装: ``` sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb sudo apt-get update sudo apt-get install cuda ``` 2. 安装cuDNN 下载对应版本的cuDNN安装包,需要注意的是,cuDNN需要与已安装的CUDA版本对应。下载完成后,解压缩文件,执行以下命令进行安装: ``` 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 /usr/local/cuda/lib64/libcudnn* ``` 安装完成后,可以通过以下命令检查cuDNN是否正确安装: ``` nvcc -V && cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 ``` 如果输出结果中显示了CUDA和cuDNN的版本信息,则说明安装成功。 3. 配置环境变量 在终端中打开.bashrc文件,在文件末尾添加以下两行: ``` export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH export PATH=/usr/local/cuda/bin:$PATH ``` 至此,cuDNN安装和配置就完成了。如果你想在深度学习框架中使用cuDNN加速,只需要将对应的库链接到项目中即可。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值