27 lines
556 B
YAML
27 lines
556 B
YAML
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 |