安装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