From 0067c17c9ab681190390dd09ba585739d761ff31 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 2 Jul 2024 08:00:41 +0000 Subject: [PATCH] update --- finetune/configure_gpu.sh | 2 ++ finetune/docker-compose_stack.yml | 2 +- finetune/docker-compose_stack1.yml | 2 +- finetune/docker-compose_stack2.yml | 2 +- 4 files changed, 5 insertions(+), 3 deletions(-) diff --git a/finetune/configure_gpu.sh b/finetune/configure_gpu.sh index b7867f1..2494aee 100755 --- a/finetune/configure_gpu.sh +++ b/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 diff --git a/finetune/docker-compose_stack.yml b/finetune/docker-compose_stack.yml index 52841bb..bc73ab0 100644 --- a/finetune/docker-compose_stack.yml +++ b/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] diff --git a/finetune/docker-compose_stack1.yml b/finetune/docker-compose_stack1.yml index cfd3b0c..cba44e2 100644 --- a/finetune/docker-compose_stack1.yml +++ b/finetune/docker-compose_stack1.yml @@ -15,7 +15,7 @@ services: reservations: generic_resources: - discrete_resource_spec: - kind: gpus + kind: NVIDIAGPU value: 1 placement: constraints: diff --git a/finetune/docker-compose_stack2.yml b/finetune/docker-compose_stack2.yml index eefd16a..4a29550 100644 --- a/finetune/docker-compose_stack2.yml +++ b/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]