From 7f295d8f08b2038857d1dc7e10b21b19d715c6a9 Mon Sep 17 00:00:00 2001
From: Your Name <you@example.com>
Date: Tue, 2 Jul 2024 06:56:33 +0000
Subject: [PATCH] =?UTF-8?q?add=20swarm-resource=E9=85=8D=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 finetune/configure_gpu.sh | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)
 mode change 100644 => 100755 finetune/configure_gpu.sh

diff --git a/finetune/configure_gpu.sh b/finetune/configure_gpu.sh
old mode 100644
new mode 100755
index 00e7c0c..b7867f1
--- a/finetune/configure_gpu.sh
+++ b/finetune/configure_gpu.sh
@@ -10,13 +10,13 @@ NODE_RESOURCES=${NODE_RESOURCES%,}  # 移除最后一个逗号
 # 生成完整的daemon.json内容
 DAEMON_JSON=$(cat <<EOF
 {
-  "default-runtime": "nvidia",
   "runtimes": {
     "nvidia": {
-      "path": "nvidia-container-runtime",
+      "path": "/usr/bin/nvidia-container-runtime",
       "runtimeArgs": []
     }
   },
+  "default-runtime": "nvidia",
   "node-generic-resources": [
     $NODE_RESOURCES
   ]
@@ -32,8 +32,12 @@ fi
 # 写入新的daemon.json内容
 echo "$DAEMON_JSON" | sudo tee /etc/docker/daemon.json > /dev/null
 
+# 添加swarm-resource配置
+sudo sed -i '/^#.*swarm-resource/s/^#//' /etc/nvidia-container-runtime/config.toml
+sudo sed -i '/swarm-resource =/s/=.*/= "DOCKER_RESOURCE_GPU"/' /etc/nvidia-container-runtime/config.toml
+
 # 重启Docker服务
-sudo systemctl restart docker
+sudo systemctl restart docker.service
 
 # 验证配置
 docker info | grep -i 'nvidia'