Ubuntu 20.04安装CUDNN

1.先根据下面连接把cuda装好

(46条消息) Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_哈希Map的博客-CSDN博客_ubuntu20.04 cuda

2.下载对应版本cudnn

CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

3.解压tgz包

4.对解压后文件夹中的文件进行拷贝

sudo cp cuda/include/cudnn* /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

5.查看cuDNN版本

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

如果输出如下,则表示安装成功:

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
首先,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加速,只需要将对应的库链接到项目中即可。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值