update

finetune/configure_gpu.sh

@@ -33,6 +33,8 @@ fi
 echo "$DAEMON_JSON" | sudo tee /etc/docker/daemon.json > /dev/null
 
 # 添加swarm-resource配置
+# swarm-resource = "DOCKER_RESOURCE_NVIDIAGPU"
+
 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
 

finetune/docker-compose_stack.yml

@@ -42,7 +42,7 @@ services:
         reservations:
           generic_resources:
             - discrete_resource_spec:
-                kind: nvidia
+                kind: NVIDIAGPU
                 value: 1
       placement:
         constraints: [node.platform.os == linux]

finetune/docker-compose_stack1.yml

@@ -15,7 +15,7 @@ services:
         reservations:
           generic_resources:
             - discrete_resource_spec:
-                kind: gpus
+                kind: NVIDIAGPU
                 value: 1
       placement:
         constraints:

finetune/docker-compose_stack2.yml

@@ -42,7 +42,7 @@ services:
         reservations:
           generic_resources:
             - discrete_resource_spec:
-                kind: nvidia
+                kind: NVIDIAGPU
                 value: 1
       placement:
         constraints: [node.labels.gpu == true]