如何构建含CUDA 11.7的Docker镜像？升级遇镜像找不到错误求解决

问题原因 #

你遇到的错误是因为NVIDIA官方并未提供nvidia/cuda:11.7-cudnn8-devel-ubuntu18.04这个组合的预构建镜像。CUDA版本与Linux发行版版本存在对应支持关系，CUDA 11.7已不再支持Ubuntu 18.04，仅提供Ubuntu 20.04、22.04等较新版本的预构建镜像。

解决方案 #

方案1：更换Ubuntu版本为20.04（推荐） #

将Dockerfile中的Ubuntu版本从18.04改为20.04，这是最简便的解决方式，修改后的Dockerfile起始代码如下：

ARG CUDA="11.7"  
ARG CUDNN="8"
FROM nvidia/cuda:${CUDA}-cudnn${CUDNN}-devel-ubuntu20.04

方案2：使用指定具体CuDNN版本的镜像（若需保留Ubuntu 18.04） #

如果你必须基于Ubuntu 18.04构建，需要指定与CUDA 11.7兼容的具体CuDNN版本号，而非仅写大版本8。例如，对应CUDA 11.7的CuDNN 8.5.0.96版本存在Ubuntu 18.04的镜像：

ARG CUDA="11.7"
ARG CUDNN="8.5.0.96"
FROM nvidia/cudnn:${CUDNN}-cuda${CUDA}-devel-ubuntu18.04

注意：这种组合的镜像选择范围很小，后续维护可能受限，优先推荐方案1。

内容的提问来源于stack exchange，提问作者Deep Learner