add swarm-resource配置

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'