diff --git a/itgpt-setup.sh b/itgpt-setup.sh index fa79fff..bcdbc30 100755 --- a/itgpt-setup.sh +++ b/itgpt-setup.sh @@ -1,6 +1,7 @@ #!/bin/bash set -euxo pipefail +Docker_urls = "https://mirrors.aliyun.com/docker-ce" export DEBIAN_FRONTEND=noninteractive sudo dpkg --set-selections <<< "cloud-init install" || true @@ -159,7 +160,7 @@ else ;; *) - echo "This version of Debian is not supported in this script." + echo "此脚本不支持此版本的 Debian。" exit 1 ;; esac @@ -200,7 +201,7 @@ fi # Check if Docker is installed if command -v docker &>/dev/null; then - echo "Docker is already installed." + echo "Docker 已安装。" else echo "未安装 Docker。正在进行安装..." # Install Docker-ce keyring @@ -211,12 +212,14 @@ else if [ -f "$FILE" ]; then sudo rm "$FILE" fi - curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o "$FILE" + + # curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o "$FILE" + curl -fsSL $Docker_urls/linux/ubuntu/gpg | sudo gpg --dearmor -o "$FILE" sudo chmod a+r /etc/apt/keyrings/docker.gpg # Add Docker-ce repository to Apt sources and install echo \ - "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu \ + "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] $Docker_urls/linux/ubuntu \ $(. /etc/os-release; echo "$VERSION_CODENAME") stable" | \ sudo tee /etc/apt/sources.list.d/docker.list > /dev/null sudo apt update -y @@ -225,9 +228,9 @@ fi # Check if docker-compose is installed if command -v docker-compose &>/dev/null; then - echo "Docker-compose is already installed." + echo "Docker-compose 已安装。" else - echo "Docker-compose is not installed. Proceeding with installations..." + echo "未安装 Docker-compose。正在进行安装..." # Install docker-compose subcommand sudo apt -y install docker-compose-plugin @@ -238,15 +241,16 @@ fi # Test / Install nvidia-docker if [[ ! -z "$NVIDIA_PRESENT" ]]; then if sudo docker run --gpus all nvidia/cuda:11.0.3-base-ubuntu18.04 nvidia-smi &>/dev/null; then - echo "nvidia-docker is enabled and working. Exiting script." + echo "nvidia-docker 已启用并正常工作。退出脚本。" else - echo "nvidia-docker does not seem to be enabled. Proceeding with installations..." + 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 sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit sudo systemctl restart docker sudo docker run --gpus all nvidia/cuda:11.0.3-base-ubuntu18.04 nvidia-smi + fi fi sudo apt-mark hold nvidia* libnvidia* @@ -266,16 +270,26 @@ echo "根据 https://github.com/NVIDIA/nvidia-docker/issues/1730 应用NVIDIA Do # Edit the Docker daemon configuration. sudo bash -c 'cat < /etc/docker/daemon.json { - "runtimes": { - "nvidia": { - "path": "nvidia-container-runtime", - "runtimeArgs": [] - } - }, - "exec-opts": ["native.cgroupdriver=cgroupfs"] + "exec-opts": [ + "native.cgroupdriver=cgroupfs" + ], + "registry-mirrors": [ + "https://hub-mirror.c.163.com", + "https://docker.m.daocloud.io", + "https://ghcr.io", + "https://mirror.baidubce.com", + "https://docker.nju.edu.cn" + ], + "runtimes": { + "nvidia": { + "path": "nvidia-container-runtime", + "runtimeArgs": [] + } + } } EOF' # Restart Docker to apply changes. sudo systemctl restart docker +sudo docker rmi nvidia/cuda:11.0.3-base-ubuntu18.04 echo "已应用解决方法。Docker 已配置为使用“cgroupfs”作为 cgroup 驱动程序。" \ No newline at end of file