debian下的运行脚本

0xSoul

1. #!/bin/bash
   
2. 
   
3. set -euxo pipefail
   
4. 
   
5. export DEBIAN_FRONTEND=noninteractive
   
6. sudo dpkg --set-selections <<< "cloud-init install" || true
   
7. 
   
8. #  设置全局变量
   
9.     # 检测系统
   
10.         OS="$(uname)"
    
11.         case $OS in
    
12.             "Linux")
    
13.                 # 判断下系统发行版
    
14.                 if [ -f /etc/os-release ]; then
    
15.                     . /etc/os-release
    
16.                     DISTRO=$ID
    
17.                     VERSION=$VERSION_ID
    
18.                 else
    
19.                     echo "Linux 发行版不受支持。"
    
20.                     exit 1
    
21.                 fi
    
22.                 ;;
    
23.         esac
    
24. 
    
25. # 检测是否存在 Nvidia GPU
    
26. NVIDIA_PRESENT=$(lspci | grep -i nvidia || true)
    
27. 
    
28. # 如果检测到 Nvidia 设备，则继续执行 Nvidia 相关步骤
    
29. if [[ -z "$NVIDIA_PRESENT" ]]; then
    
30.     echo "NVIDIA驱动不存在，准备安装."
    
31. else
    
32. # 查看nvidia-smi是否正常工作
    
33.     if command -v nvidia-smi &>/dev/null; then
    
34.         echo "CUDA drivers 已经安装 nvidia-smi 工作正常."
    
35.     else
    
36. 
    
37.         case $DISTRO in
    
38.             "debian")
    
39.                 case $VERSION in
    
40.                     "10"|"11")
    
41.                         # 支持 Debian 10 & 11
    
42.                         sudo -- sh -c 'apt update; apt upgrade -y; apt autoremove -y; apt autoclean -y'
    
43.                         sudo apt install linux-headers-$(uname -r) -y
    
44.                         sudo apt install linux-hwe  -y
    
45.                         sudo apt install linux-modules-$(uname -r)  -y
    
46.                         sudo apt install linux-modules-extra-$(uname -r)  -y
    
47.                         # 重新安装网卡驱动
    
48.                         sudo apt-get install r8168-dkms -y
    
49. 
    
50.                         sudo apt update -y
    
51.                         sudo apt install nvidia-driver firmware-misc-nonfree
    
52.                         wget https://developer.download.nvidia.com/compute/cuda/repos/debian${VERSION}/x86_64/cuda-keyring_1.1-1_all.deb
    
53.                         sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit
    
54.                         sudo apt update -y
    
55.                         ;;
    
56. 
    
57.                     *)
    
58.                         echo "debian 版本不被支持"
    
59.                         exit 1
    
60.                         ;;
    
61.                 esac
    
62.                 ;;
    
63. 
    
64.             *)
    
65.                 echo "只搞 debian 不干其它，换个系统."
    
66.                 exit 1
    
67.                 ;;
    
68.         esac
    
69.         echo " 5 秒后，即将 reboot !!! 重启后，再干一遍这个脚本，确保驱动完整 !"
    
70.         sleep 5s
    
71.         sudo reboot
    
72.     fi
    
73. fi
    
74. #  测试驱动
    
75. if [[ ! -z "$NVIDIA_PRESENT" ]]; then
    
76.     nvidia-smi
    
77. fi
    
78. 
    
79. # 查看 docker 是否安装
    
80. if command -v docker &>/dev/null; then
    
81.     echo "Docker 已经安装."
    
82. else
    
83.     echo "Docker 没有安装..."
    
84.     # 安装 Docker-ce keyring
    
85.     sudo apt update -y
    
86.     sudo apt install -y ca-certificates curl gnupg
    
87.     sudo install -m 0755 -d /etc/apt/keyrings
    
88.     FILE=/etc/apt/keyrings/docker.gpg
    
89.     if [ -f "$FILE" ]; then
    
90.         sudo rm "$FILE"
    
91.     fi
    
92.     curl -fsSL https://download.docker.com/linux/debian/gpg | sudo gpg --dearmor -o "$FILE"
    
93.     sudo chmod a+r /etc/apt/keyrings/docker.gpg
    
94. 
    
95.     # 添加 Docker-ce repository 到 Apt sources 并且 install
    
96.     echo \
    
97.       "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/debian \
    
98.       $(. /etc/os-release; echo "$VERSION_CODENAME") stable" | \
    
99.       sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
    
100.     sudo apt update -y
     
101.     sudo apt -y install docker-ce
     
102. fi
     
103. 
     
104. # 检查 docker-compose是否安装
     
105. if command -v docker-compose &>/dev/null; then
     
106.     echo "Docker-compose 存在."
     
107. else
     
108.     echo "Docker-compose 不存在. 干它一波..."
     
109. 
     
110.     # 安装 docker-compose subcommand
     
111.     sudo apt -y install docker-compose-plugin
     
112.     sudo ln -sv /usr/libexec/docker/cli-plugins/docker-compose /usr/bin/docker-compose
     
113.     docker-compose --version
     
114. fi
     
115. 
     
116. # Test / 安装 nvidia-docker
     
117. if [[ ! -z "$NVIDIA_PRESENT" ]]; then
     
118.     if sudo docker run --gpus all nvidia/cuda:11.0.3-base-ubuntu18.04 nvidia-smi &>/dev/null; then
     
119.         echo "nvidia-docker 开启且已经正常工作"
     
120.     else
     
121.         echo "nvidia-docker没工作. 准备安装..."
     
122.         distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
     
123.         curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add
     
124.         curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
     
125.         sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
     
126.         sudo systemctl restart docker
     
127.         sudo docker run --gpus all nvidia/cuda:11.0.3-base-ubuntu18.04 nvidia-smi
     
128.     fi
     
129. fi
     
130. sudo apt-mark hold nvidia* libnvidia*
     
131. # 添加 docker group 和 user 到 group docker
     
132. sudo groupadd docker || true
     
133. sudo usermod -aG docker $USER || true
     
134. newgrp docker || true
     
135. # 解决 NVIDIA Docker 问题
     
136. echo "正在应用解决方案，针对 NVIDIA Docker 的问题，参考 https://github.com/NVIDIA/nvidia-docker/issues/1730"
     
137. # 问题及解决方案概要：
     
138. # 问题发生在主机执行 daemon-reload 时，这可能导致使用 systemd 的容器失去对 NVIDIA GPU 的访问。
     
139. # 要检查是否受影响，在主机上运行 `sudo systemctl daemon-reload`，然后在容器中用 `nvidia-smi` 检查 GPU 访问。
     
140. # 如果受影响，请按照以下解决方案进行处理。
     
141. 
     
142. # 解决方案步骤：
     
143. # 禁用 Docker 容器的 cgroups，以防止出现问题。
     
144. # 编辑 Docker 守护进程配置。
     
145. sudo bash -c 'cat <<EOF > /etc/docker/daemon.json
     
146. {
     
147.    "runtimes": {
     
148.        "nvidia": {
     
149.            "path": "nvidia-container-runtime",
     
150.            "runtimeArgs": []
     
151.        }
     
152.    },
     
153.    "exec-opts": ["native.cgroupdriver=cgroupfs"]
     
154. }
     
155. EOF'
     
156. 
     
157. # 重启 Docker 以应用更改。
     
158. sudo systemctl restart docker
     
159. echo "解决方案已应用。Docker 已配置为使用 'cgroupfs' 作为 cgroup 驱动。"
     
160. 
     

复制代码