rm /etc/apt/sources.list.d/nvidia-container-toolkit.list

rm /etc/apt/sources.list.d/nvidia-docker.list

distribution="ubuntu22.04"

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit.gpg

curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list \

  | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit.gpg] https://#g' \

  | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

apt-get update

apt-get install -y nvidia-docker2

systemctl restart docker

  # 验证

docker run --gpus all --rm nvidia/cuda:12.8.1-base-rockylinux8 nvidia-smi

# 安装NVIDIA-Docker（国内网络；清华大学镜像源）

mkdir -p /etc/apt/sources.list.d/

tee /etc/apt/sources.list.d/nvidia-docker.list <<EOF

deb https://mirrors.tuna.tsinghua.edu.cn/nvidia-docker/ubuntu22.04/amd64 /

EOF

curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | \

  gpg --dearmor | sudo tee /usr/share/keyrings/nvidia-container-toolkit.gpg > /dev/null

apt-get update

apt-get install -y nvidia-docker2

systemctl restart docker

  # 验证

docker run --gpus all --rm nvidia/cuda:12.8.1-base-rockylinux8 nvidia-smi