如何修改此命令来安装特定的cuDNN和torch版本“apt install libcudnn8”

Question

这是我的系统：

No LSB modules are available.
Distributor ID: Ubuntu
Description:    Ubuntu 22.04.2 LTS
Release:        22.04
Codename:       jammy

我想安装 cudnn 8.7.0 - 我认为确切的版本是 8.7.0.84_1

还有cuda11.7

此命令：

apt install libcudnn8

安装：

Unpacking libcudnn8-dev (8.9.2.26-1+cuda12.1) over (8.5.0.96-1+cuda11.7) ...

那么如何修改

apt install libcudnn8

以便安装

cuDNN 8.7.0

和

Cuda 11.7

？

谢谢你

Answer 1

根据 Nvidia 文档（https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#package-manager-ubuntu-install）：

sudo apt-get install libcudnn8=${cudnn_version}-1+${cuda_version}
sudo apt-get install libcudnn8-dev=${cudnn_version}-1+${cuda_version}
sudo apt-get install libcudnn8-samples=${cudnn_version}-1+${cuda_version}

地点：

• ${cudnn_version}

  例如 8.9.2.*

• ${cuda_version}

  例如 cuda12.1 或 cuda11.8

您可以通过运行以下命令找到可安装版本的列表：

sudo apt update
sudo apt list -a libcudnn8-dev

然后将上面命令中的变量替换为您需要的命令。

对于 RunPod，您已登录到 docker 容器，并且未安装

sudo

命令，因此您显然可以删除

sudo

前缀，因为您无论如何都以

root

用户身份登录，因此对于您的特定用例RunPod 的命令是：

apt install libcudnn8=8.7.0.84-1+cuda11.8 libcudnn8-dev=8.7.0.84-1+cuda11.8

注意：虽然您特别指定了 CUDA 11.7，但没有可用的软件包来满足该特定要求，这就是选择 CUDA 11.8 的原因。

Answer 2

并非所有 cuDNN 和 CUDA 版本都可用，即使您安装了 12.2，nvidia-smi 中的 CUDA 版本也可能显示为 12.0。

因此，例如，下面的代码不起作用：

# doesn't work
sudo apt-get install libcudnn8=8.9.4.25-1+cuda12.0

但这有效：

sudo apt-get install libcudnn8=8.9.4.25-1+cuda12.2

要查看可用的 libcudnn8 版本，请参阅下文： https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/