安装cuDNN

安装

我们参考Installing cuDNN on Linux进行安装，我们可以从cudnn下载

我的系统是Ubuntu 22.04 X64版本，所以我选择Local Installer for Ubuntu22.04 x86_64 (Deb)

依次执行

ruby 复制代码

thinktik@thinkstation:~/Downloads$ sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.2.26_1.0-1_amd64.deb 
thinktik@thinkstation:~/Downloads$ sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
thinktik@thinkstation:~/Downloads$ sudo apt-get update

ruby 复制代码

thinktik@thinkstation:~/Downloads$ sudo apt-get install libcudnn8
thinktik@thinkstation:~/Downloads$ sudo apt-get install libcudnn8-dev
thinktik@thinkstation:~/Downloads$ sudo apt-get install libcudnn8-samples

校验

ruby 复制代码

thinktik@thinkstation:~/Downloads$ cp -r /usr/src/cudnn_samples_v8/ $HOME
thinktik@thinkstation:~/Downloads$ cd  $HOME/cudnn_samples_v8/mnistCUDNN
thinktik@thinkstation:~/cudnn_samples_v8/mnistCUDNN$ make clean && make
# 如果没报错就没必要安装
thinktik@thinkstation:~/cudnn_samples_v8/mnistCUDNN$ sudo apt-get install libfreeimage3 libfreeimage-dev

执行检测，看到pass表示测试通过

ruby 复制代码

thinktik@thinkstation:~/cudnn_samples_v8/mnistCUDNN$ ./mnistCUDNN