diff --git a/finetune/docker-compose_stack.yml b/finetune/docker-compose_stack.yml index 5fecd98..52841bb 100644 --- a/finetune/docker-compose_stack.yml +++ b/finetune/docker-compose_stack.yml @@ -43,7 +43,7 @@ services: generic_resources: - discrete_resource_spec: kind: nvidia - value: 8 + value: 1 placement: constraints: [node.platform.os == linux] cap_add: diff --git a/finetune/docker-compose_stack1.yml b/finetune/docker-compose_stack1.yml new file mode 100644 index 0000000..b8e4497 --- /dev/null +++ b/finetune/docker-compose_stack1.yml @@ -0,0 +1,27 @@ +version: '3.8' + +services: + ubuntu-finetune: + image: hotwa/deepspeed:pt23 + ports: + - 3228:22 + environment: + - NVIDIA_VISIBLE_DEVICES=all + - NVIDIA_DRIVER_CAPABILITIES=compute,utility + - TMPDIR=/var/tmp + deploy: + replicas: 1 + resources: + reservations: + generic_resources: + - discrete_resource_spec: + kind: gpus + value: 1 + placement: + constraints: [node.platform.os == linux] + cap_add: + - IPC_LOCK + +networks: + default: + driver: overlay \ No newline at end of file