====== Nvidia的GPU直接在Docker容器内部启动的方法 ====== =====如果你有科学上网的方式==== 需要在宿主机上执行如下操作【注意:此部分需要科学上网,不然无法访问nvidia.github.io 】: # 安装依赖 sudo apt-get update sudo apt-get install -y curl # 添加密钥和仓库 curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \ sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \ sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list sudo apt-get update sudo apt-get install -y nvidia-container-toolkit # 生成配置文件 sudo nvidia-ctk runtime configure --runtime=docker # 重启Docker服务 sudo systemctl restart docker 如果成功后,可以用Docker进行测试 ''docker run --rm --gpus all nvidia/cuda:12.6.0-base-ubuntu24.04 nvidia-smi'' 如果失败,会输出对应的失败文案。 自己动手启动Docker容器的时候,记得带上 --gpus all 参数,不然Docker容器里仍旧无法使用GPU ===== 如果你没有科学上网;但是选择相信Ctbots.com 我们提供加速方式 ===== 步骤一:安装 nVidia密钥 curl -fsSL https://748996.xyz/nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg 步骤二:增加nVidia的下载deb仓库 sudo vim /etc/apt/sources.list.d/nvidia-container-toolkit.list 增加如下内容: deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://748996.xyz/nvidia.github.io/libnvidia-container/stable/deb/$(ARCH) / 步骤三: sudo apt-get update sudo apt-get install -y nvidia-container-toolkit 步骤四:重新配置Docker # 生成配置文件 sudo nvidia-ctk runtime configure --runtime=docker # 重启Docker服务 sudo systemctl restart docker 步骤五:测试Docker的GPU直通 docker run --rm --gpus all nvidia/cuda:12.6.0-base-ubuntu24.04 nvidia-smi