-
Notifications
You must be signed in to change notification settings - Fork 17
/
Copy pathdocker-compose-local.yaml
73 lines (67 loc) · 1.84 KB
/
docker-compose-local.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
---
services:
eng-ai-agents:
build:
context: "."
dockerfile: ./docker/Dockerfile.dev.gpu
image: eng-ai-agents
shm_size: '8gb' # Shared memory size
user: vscode
command: /bin/sh -c "while sleep 1000; do :; done"
# depends_on:
# - minio
#- postgresml
environment:
- NVIDIA_VISIBLE_DEVICES=all # or specify GPUs as 0,1 for specific devices
deploy:
resources:
reservations:
devices:
- driver: nvidia
capabilities: ["gpu"]
device_ids: ["all"] # Specify GPU IDs if needed, or 'all' to use all available GPUs
networks:
- eng-ai-agents-net
# minio:
# image: minio/minio:latest
# container_name: minio
# ports:
# - "9000:9000"
# - "9001:9001" # MinIO Console
# environment:
# MINIO_ROOT_USER: minioadmin
# MINIO_ROOT_PASSWORD: minioadmin
# command: server /data --console-address ":9001"
# volumes:
# - minio_data:/data
# networks:
# - eng-ai-agents-net
# healthcheck:
# test: ["CMD-SHELL", "curl --fail http://localhost:9000/minio/health/live || exit 1"]
# postgresml:
# image: ghcr.io/postgresml/postgresml:2.9.3
# command: >
# sh -c "sudo -u postgresml psql -d postgresml"
# volumes:
# - postgresml_data:/var/lib/postgresql
# expose:
# - 54320
# ports:
# - "54320:5432"
# - "8000:8000"
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# capabilities: ["gpu"]
# device_ids: ["all"] # Specify GPU IDs if needed, or 'all' to use all available GPUs
# stdin_open: true
# tty: true
networks:
eng-ai-agents-net:
driver: bridge
volumes:
# minio_data:
#postgresml_data:
rye-cache: