使用apt命令在Ubuntu上安装Cudnn

2 年 ago

科, 颖

1 minute

(2023/11/23补充)
由于没有针对Ubuntu22.04的machine-learning/repos/ubuntu2204，因此以下方法无法使用，但不知何时就可以直接安装CUDA，并且能够安装libcudnn等。

通过添加以下NVIDIA DIGITS安装方法说明的repo，似乎可以使用apt安装libcudnn，虽然不知道为什么会这样。
https://github.com/NVIDIA/DIGITS/blob/master/docs/BuildDigits.md

只需执行以下操作即可。

$ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add -
$ echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
$ apt-get update
$ apt-cache search libcudnn
libcudnn5-dev - cuDNN development libraries and headers
libcudnn5 - cuDNN runtime libraries
libcudnn6-dev - cuDNN development libraries and headers
libcudnn6 - cuDNN runtime libraries
libcudnn7-dev - cuDNN development libraries and headers
libcudnn7 - cuDNN runtime libraries
libcudnn7-doc - cuDNN documents and samples

虽然有 Ver Up 出来，但是通过 apt-get upgrade 可以轻松处理。
为什么 cudnn 官方安装方法中没有提到这一点呢？