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