Ubuntu16.04安装cudnn5.1

最新推荐文章于 2021-09-25 21:43:43 发布

转载最新推荐文章于 2021-09-25 21:43:43 发布 · 1k 阅读

文章标签：

#Ubuntu #cudnn

CUDA 专栏收录该内容

1 篇文章

订阅专栏

本文详细介绍了如何从NVIDIA官方网站下载cuDNN，并提供了适用于Linux系统的安装步骤。包括解压文件、复制头文件、创建软链接及更新链接库等关键操作。

获取cuDNN

去官网https://developer.nvidia.com/rdp/cudnn-download下载cuDNN，要注册NVIDIA账号。

我选择的是cuDNN v5.1 -> cuDNN v5.1 Library for Linux

安装cuDNN

下载cuDNN后解压

cd cuda/include

sudo cp cudnn.h /usr/local/cuda/include #复制头文件

cd ../lib64

sudo rm -rf libcudnn.so libcudnn.so.5 #删除原动态文件

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软链接

sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

若需要更换cudnn版本，则替换原来的libcudnn*，并重新软链接。

更新链接库：

sudo ldconfig

另外一个版本

从官网下载cudnn-8.0-linux-x64-v5.1.tgz for CUDA 8.0. 解压到当前目录：

tar -zxvf cudnn-8.0-linux-x64-v5.1.tgz

解压后的文件如下：

cuda/include/cudnn.h
cuda/lib64/libcudnn.so
cuda/lib64/libcudnn.so.5
cuda/lib64/libcudnn.so.5.1.5
cuda/lib64/libcudnn_static.a

然后执行：

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*