如何为Docker镜像同时配置Java JAR运行环境与NVIDIA CUDA依赖

最优实现方案：CUDA基础镜像安装OpenJDK #

优先选择你提出的第二种思路，稳定性和实现成本远优于在OpenJDK镜像中装CUDA的方案。nvidia官方提供的cuda基础镜像已经完成了CUDA底层依赖、环境变量、库路径的所有配置，不存在版本适配问题，仅需额外安装JDK即可满足需求，操作步骤如下：

参考Dockerfile配置 #

# 选用你之前验证过的可用CUDA基础镜像
FROM nvidia/cuda:10.2-base-ubuntu18.04
# 禁用交互提示
ENV DEBIAN_FRONTEND=noninteractive
# 安装基础工具
RUN apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    curl \
    gnupg2 \
    software-properties-common \
    && rm -rf /var/lib/apt/lists/*
# 添加OpenJDK第三方源（Ubuntu18.04官方源无openjdk-14安装包）
RUN add-apt-repository ppa:openjdk-r/ppa
# 安装OpenJDK14
RUN apt-get update && apt-get install -y --no-install-recommends \
    openjdk-14-jdk \
    && rm -rf /var/lib/apt/lists/*
# 可选：安装Python3、PyTorch等你需要的其他依赖
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3 \
    python3-pip \
    && rm -rf /var/lib/apt/lists/* \
    && pip3 install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

环境验证 #

镜像构建完成后执行以下命令验证功能正常：

• 验证Java环境：java -version，输出包含14版本信息即正常
• 验证PyTorch GPU可用性：python3 -c "import torch;print(torch.cuda.is_available())"，输出True即正常

其他可选方案 #

如果不需要自己搭建PyTorch环境，可以直接用PyTorch官方预构建的GPU镜像作为基础，仅需在镜像中安装OpenJDK14即可，比从CUDA基础镜像构建更省步骤，基础镜像可选用pytorch/pytorch:1.8.1-cuda10.2-cudnn7-runtime。

注意事项 #

• 运行容器时必须添加GPU参数：使用nvidia-docker run或者普通docker命令添加--gpus all参数，否则容器内无法识别GPU设备
• 宿主机NVIDIA驱动版本需要匹配CUDA版本要求，CUDA10.2要求宿主机驱动版本不低于440.33

不推荐第一种思路的原因 #

在OpenJDK:buster（Debian10）基础镜像中安装CUDA需要手动添加NVIDIA的Debian源、匹配CUDA版本和系统依赖、手动配置CUDA_HOME、LD_LIBRARY_PATH等环境变量，操作步骤繁琐，非常容易出现依赖冲突、版本不匹配导致PyTorch无法识别GPU的问题，无特殊需求不建议选用。 #

内容的提问来源于stack exchange，提问作者Ramraj Chandradevan