From bc13f60951a1f6f04d86ff1c8c7a49ae9bb0a3ae Mon Sep 17 00:00:00 2001
From: hyh <2667004988@qq.com>
Date: Wed, 12 Jun 2024 12:52:42 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0nvidia=20=20docker=20containe?=
 =?UTF-8?q?r?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 itgpt-setup.sh | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/itgpt-setup.sh b/itgpt-setup.sh
index 9a21c62..5c8402a 100755
--- a/itgpt-setup.sh
+++ b/itgpt-setup.sh
@@ -247,8 +247,11 @@ if [[ ! -z "$NVIDIA_PRESENT" ]]; then
     else
         echo "nvidia-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
+        # 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
+        # nvidia.github.io <- nvidia-docker.geekery.cn
+        curl -s -L https://nvidia-docker.geekery.cn/nvidia-docker/gpgkey | sudo apt-key add
+        curl -s -L https://nvidia-docker.geekery.cn/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
         sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
         sudo systemctl restart docker 
         sudo docker run --rm  --gpus all   $images nvidia-smi