Ubuntu cuDNN 安装


命令:

注意先从官网上下载对应的 cudnn 版本!
注意 cudnn9.0 刚刚发布,我们需要的是 cudnn 8.x 和 cudnn 7,不要搞混了。

以 v8.9.7 ubuntu 20.04 为例(其余版本类似)

下载 deb 安装包:

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb

sudo apt-get update

sudo apt-get install libcudnn8
sudo apt-get install libcudnn8-dev

注意版本号匹配即可

查看安装位置:


sudo find / -name libcudnn*

安装成功测试:

Copy the cuDNN samples to a writable path.

cp -r /usr/src/cudnn_samples_v8/ $HOME

Go to the writable path.

cd  $HOME/cudnn_samples_v8/mnistCUDNN

Compile the mnistCUDNN sample.

make clean && make

Run the mnistCUDNN sample.

./mnistCUDNN

If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:

Test passed!

另一种安装方法是下载 .tar 文件,然后依次运行:


tar -xzvf

sudo cp cuda/include/*    /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*

对于普通文件而言,不加 -r 选项也是可以的。cp 命令会默认复制文件而不会复制目录。但如果 cuda/include/ 目录下有子目录,且你也想复制子目录及其内容,那么就需要使用 -r(或 --recursive)选项。如果目录下只有文件而没有子目录,则不加 -r 选项也是可以的,cp 命令会将所有文件复制到目标目录中。


Author: Yixiang Zhang
Reprint policy: All articles in this blog are used except for special statements CC BY 4.0 reprint policy. If reproduced, please indicate source Yixiang Zhang !
评论
  TOC