Ubuntu 22.04/24.04 64位 预装NVIDIA GPU驱动镜像是专为AI开发与深度学习场景优化的高性能公共镜像,预置了主流版本的NVIDIA GPU驱动、CUDA、Docker引擎及NVIDIA Container Toolkit 等,开箱即用,快速部署GPU容器化环境,运行大模型训练、推理任务,显著简化底层依赖配置,提升AI应用开发与部署效率。
预配置软件信息
该公共镜像预置了最新版本的Docker及性能调优软件keentune,并支持OFED和eRDMA,其他预置驱动及软件信息如下:
内核版本、驱动及软件信息 | Ubuntu 22.04 64位 预装NVIDIA GPU驱动镜像 | Ubuntu 24.04 64位 预装NVIDIA GPU驱动镜像 |
操作系统内核版本 | 5.15.0-161-generic | 6.8.0-87-generic |
Nvidia GPU 驱动版本 | 570.133.20 | 570.133.20 |
NVIDIA Container Toolkit | 1.17.8 | 1.17.8 |
CUDA版本 | 12.8 | 12.8 |
cuDNN版本 | 9.8.0.87 | 9.8.0.87 |
NCCL | v2.27.7-1 | v2.27.7-1 |
OpenMPI | 4.1.3 | 4.1.3 |
Python 3 | 3.10.12 | 3.12.3 |
历史版本镜像的驱动及软件信息请查看Ubuntu公共镜像发布记录。
Ubuntu 22.04 64位 预装NVIDIA GPU驱动镜像
支持的规格族
gn7e、gn7s、gn7i、gn6v、gn6i、gn6e、gn5、gn5i
ebmgn7e、ebmgn7i、ebmgn6v、ebmgn6i、ebmgn6e
ebmgn7ix、ebmgn7ex
gn8is、ebmgn8is、gn8v、ebmgn8v
系统环境变量配置
/etc/profile.d/nccl_mpi.sh
export NCCL_HOME=/usr/local/nccl export MPI_HOME=/usr/local/openmpi export LD_LIBRARY_PATH=${NCCL_HOME}/lib:${MPI_HOME}/lib:$LD_LIBRARY_PATH export PATH=${MPI_HOME}/bin:$PATH/etc/profile.d/cuda.sh
export PATH=/usr/local/cuda/bin:$PATH export CUDA_HOME=/usr/local/cuda export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
Ubuntu 24.04 64位 预装NVIDIA GPU驱动镜像
支持的规格族
gn7e、gn7s、gn7i、gn6v、gn6i、gn6e、gn5、gn5i
ebmgn7e、ebmgn7i、ebmgn6v、ebmgn6i、ebmgn6e
ebmgn7ix、ebmgn7ex
gn8is、ebmgn8is、gn8v、ebmgn8v
系统环境变量配置
/etc/profile.d/nccl_mpi.sh
export NCCL_HOME=/usr/local/nccl export MPI_HOME=/usr/local/openmpi export LD_LIBRARY_PATH=${NCCL_HOME}/lib:${MPI_HOME}/lib:$LD_LIBRARY_PATH export PATH=${MPI_HOME}/bin:$PATH/etc/profile.d/cuda.sh
export PATH=/usr/local/cuda/bin:$PATH export CUDA_HOME=/usr/local/cuda export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH