深度学习 Docker 镜像怎么选：CUDA、框架与架构的三层匹配

难度：入门。本文为中文技术整理，保留来源中的技术边界；示例只用于阅读，未在本服务器执行。

宿主机驱动、容器 CUDA 用户态库和 PyTorch/TensorFlow 构建版本是三层依赖。先确认 GPU 架构与驱动，再选 CUDA 基线，最后选框架镜像。Jetson 还必须围绕 JetPack/L4T 选择。

示例

print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.is_available())

来源

• https://docs.docker.com/compose/how-tos/gpu-support/