ubuntu - ubuntu20.04 docker容器中NVIDIA cuDNN安装

Question

我尝试按照 official guide 安装 NVIDIA cuDNN 8.3.2在 Dockerfile 中:

ENV OS=ubuntu2004

RUN wget https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin

RUN mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/7fa2af80.pub
RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/ /"
RUN apt-get update

ENV cudnn_version=8.3.2.1
ENV cuda_version=cuda11.5

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

我收到错误: E:未找到“libcudnn8”的版本“8.3.2.1-1+cuda11.5”命令“/bin/bash -o pipefail -c apt-get install libcudnn8=${cudnn_version}-1+$” {cuda_version}'返回非零代码:100

我访问推荐support matrix对于版本兼容性，但遗憾的是没有找到合适的说明。有人知道这里需要哪个 cudnn 和 cuda 版本吗？是否有指向 REPO 的链接，我可以在其中查看可用的软件包，like this one ？

Answer 1

我在互联网上为开发者的官方网站上找到了必要的存储库:

cuda 不同操作系统的存储库:https://developer.download.nvidia.com/compute/cuda/repos/

不同 cuda 版本的

libcudnn8 存储库: https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/

cuda 和 cuDNN 的 onnx 执行提供程序兼容性矩阵:https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html

所以，就我而言，答案是:

ENV cudnn_version=8.2.4.15
ENV cuda_version=cuda11.4