命令:
注意先从官网上下载对应的 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 命令会将所有文件复制到目标目录中。