最稳妥方案是直接使用 NVIDIA NGC 官方镜像构建，优先选用预编译的 nvcr.io/nvidia/pytorch 或 tensorflow 镜像，基于其轻量扩展定制组件，确保宿主机驱动≥525、正确配置 nvidia-container-toolkit，并用 --gpus all 运行容器，最后通过框架级代码验证 GPU 可用性。
直接用 NVIDIA NGC 官方镜像构建最稳妥，不建议从零手写基础环境。关键在选对底镜、正确扩展、验证 GPU 可用性，而非重复造轮子。
选对基础镜像：优先使用 NGC 预编译镜像
NGC（NVIDIA GPU Cloud）提供的 nvcr.io/nvidia/pytorch 或 nvcr.io/nvidia/tensorflow 镜像已预装匹配版本的 CUDA 运行时、cuDNN、驱动兼容层和主流框架，且经过 NVIDIA 官方测试。例如：
-
nvcr.io/nvidia/pytorch:23.10-py3对应 CUDA 12.2、PyTorch 2.1 -
nvcr.io/nvidia/tensorflow:23.12-tf2-py3对应 CUDA 12.3、TensorFlow 2.15
这些镜像内已设好 CUDA_HOME、LD_LIBRARY_PATH 等关键变量，无需手动配置路径或安装驱动。
定制化构建：基于 NGC 镜像做轻量扩展
若需添加 detectron2、wandb、custom ops 等组件，应在 NGC 镜像基础上扩展，而非从 ubuntu 或 anaconda3 开始构建：
- Dockerfile 第一行必须为
FROM nvcr.io/nvidia/pytorch:23.10-py3 - 禁用
ENV CUDA_HOME=...类覆盖操作，避免破坏原有 CUDA 路径 - 用
pip install --no-cache-dir -r requirements.txt减小镜像体积 - 如需 conda 包，先运行
conda init bash && source ~/.bashrc再调用conda install
构建与运行时的关键参数
构建后运行容器时，GPU 支持依赖宿主机配置与命令参数协同生效：
- 宿主机必须已安装 ≥525 版本 NVIDIA 驱动，并通过
nvidia-smi验证 - 已安装
nvidia-container-toolkit并执行sudo nvidia-ctk runtime configure --runtime=docker - 运行命令必须含
--gpus all，例如：docker run --gpus all -it --shm-size=8g my-dl-env - 对某些框架（如 PyTorch + custom kernel），可追加环境变量增强兼容：
-e NVIDIA_DRIVER_CAPABILITIES=compute,utility
验证是否真正启用 GPU
进入容器后，不能只看 nvidia-smi，要确认深度学习框架实际调用 GPU：
- PyTorch：
python -c "import torch; print(torch.cuda.is_available(), torch.cuda.device_count())" - TensorFlow：
python -c "import tensorflow as tf; print(len(tf.config.list_physical_devices('GPU')))" - 若返回
True和设备数 > 0，说明 GPU 已被框架识别并可用
