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
    评论
Ubuntu 22.04(Focal Fossa)上安装CUDA和cuDNN通常涉及几个步骤。CUDA是NVIDIA的并行计算平台和SDK,而cuDNN是用于深度学习的GPU加速库。以下是安装CUDA和cuDNN的基本流程: 1. **安装CUDA:** - 首先,确保你已经访问了NVIDIA的开发者网站并下载了适用于Ubuntu的CUDA安装包,通常选择与你系统GPU型号匹配的版本。 - 在终端中运行命令(假设下载的文件名是cuda-repo-ubuntu2204-<version>.deb): ``` sudo dpkg -i cuda-repo-ubuntu2204_<version>.deb sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/repos/ubuntu/focal/cpp sudo apt update sudo apt install cuda-cudart_<version> -y ``` 2. **安装cuDNN:** - NVIDIA也提供了cuDNN的.deb包,你可以从NVIDIA Developer Zone下载cuDNN的相应版本。 - 将cuDNN文件添加到软件源: ``` sudo mv <cudnn_file>.deb /var/cuda-repo-ubuntu2204/ sudo dpkg -i /var/cuda-repo-ubuntu2204/<cudnn_file>.deb sudo apt-key adv --fetch-keys file:///var/cuda-repo-ubuntu2204/7fa2af80.pub sudo apt update sudo apt install libcudnn8_<version> -y ``` 注意替换`<cudnn_file>`为实际下载的cuDNN包的名称。 3. **配置环境变量:** - 添加CUDA路径到系统PATH环境变量: ```bash echo "export PATH=/usr/local/cuda-<version>/bin:$PATH" >> ~/.bashrc source ~/.bashrc ``` 4. **验证安装:** - 运行`nvidia-smi`检查GPU状态,`cudnnVersion`命令查看cuDNN是否安装成功。 **相关问题:** 1. CUDA的官网地址是多少? 2. 我应该根据我的GPU型号选择哪个版本的CUDA? 3. 如何检查我的Ubuntu系统的CUDA版本?
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值