安装NVIDIA Container Toolkit.md 1.3 KB


notion-id: 23aca4d7-cb7b-808d-9918-e315643ea9

用于让 GPU 可被 Docker 容器访问:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list

#cat /etc/os-release 查看版本不支持所以不运行下载

#手动将支持的版本(如 Ubuntu 22.04)写入配置,以绕过限制。
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu22.04/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt update
sudo apt install -y nvidia-docker2
# 配置 Docker 默认 runtime
sudo tee /etc/docker/daemon.json <<EOF
{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}
EO
# 重启 Docker 服务
sudo systemctl restart docker

(4) 验证 GPU 可用性

执行:

docker run --rm --gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi

若输出 GPU 型号和信息 ✅,即可用于 BLIP 模型部署。