CUDNN安装及版本选择

CUDNN下载地址
在这里插入图片描述
选择箭头所指版本进行下载并解压。

解压完成后,打开终端

cd cudnn-10.2-linux-x64-v8.1.1.33/cuda/

	sudo cp include/cudnn.h /usr/local/cuda-10.2/include/
	sudo cp lib64/libcudnn* /usr/local/cuda-10.2/lib64/
	sudo cp include/cudnn_version.h /usr/local/cuda-10.2/include/
	sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
	sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*

注意自己cuda的版本
验证

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

修改cudnn软连接

cd /usr/local/cuda-10.2/lib64/
sudo rm libcudnn.so.8
sudo ln -s libcudnn.so8.1.1 libcudnn.so.8

上边命令要对应自己的版本号进行修改。例如,如果你是cudnn版本是7.6.5,那么命令就是

sudo rm libcudnn.so.7
sudo ln -s libcudnn.so.7.6.5 libcudnn.so.7
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
您可以按照以下步骤下载并安装CUDNN: 1. 登录到NVIDIA开发者网站:https://developer.nvidia.com/。 2. 在顶部菜单中选择 "登录",使用您的NVIDIA帐户登录。如果没有帐户,您可以在网站上注册一个新帐户。 3. 登录后,将鼠标悬停在顶部菜单上的 "下载" 上,然后选择 "所有下载"。 4. 在 "操作系统" 下拉菜单中选择适合您系统的操作系统类型。 5. 在 "类别" 下拉菜单中选择 "加速库"。 6. 滚动页面并找到 "cuDNN",然后点击它。 7. 在 "cuDNN 下载" 页面上,选择适合您系统的 cuDNN 版本,并点击 "下载"。 8. 下载完成后,解压缩下载的文件。 安装CUDNN: 1. 打开终端或命令提示符,进入解压缩的文件夹。 2. 复制 cuDNN 文件到 CUDA 安装目录。例如,在 Linux 上,默认的 CUDA 安装目录是 /usr/local/cuda。您可以使用以下命令复制文件: ``` sudo cp -P include/cudnn*.h /usr/local/cuda/include sudo cp -P lib64/libcudnn* /usr/local/cuda/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn* ``` 3. 添加 cuDNN 到环境变量。编辑您的 .bashrc 文件,并在文件末尾添加以下行: ``` export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 4. 保存并关闭 .bashrc 文件,然后运行以下命令使更改生效: ``` source ~/.bashrc ``` 5. 现在,您已成功安装 cuDNN。您可以在代码中引入 cuDNN 库并使用它。 请注意,安装 cuDNN 需要先安装 CUDA,因为 cuDNN 是 CUDA 的一个加速库。在安装 cuDNN 之前,请确保已正确安装并配置了 CUDA。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值