z-image-generation/docker-compose.yml at master · KatherLab/z-image-generation · GitHub

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
version: '3.8'

services:
  z-image-webapp:
    build:
      context: .
      dockerfile: Dockerfile
    container_name: z-image-turbo-webapp
    ports:
      - "3010:3010"
    volumes:
      # Mount the data directory for storing generated images
      - /tim/data:/tim/data
      # Mount the cache directory for huggingface models
      - /tim/cache:/home/vscode/.cache
      # Mount the code directory for development
      - .:/workspace
    environment:
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - CUDA_VISIBLE_DEVICES=0
      - HF_HOME=/home/vscode/.cache/huggingface
      - XDG_CACHE_HOME=/home/vscode/.cache
      - TRANSFORMERS_CACHE=/home/vscode/.cache/huggingface
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ['0']  # GB10 GPU
              capabilities: [gpu]
    restart: unless-stopped
    shm_size: '24gb'
    networks:
      - z-image-network

networks:
  z-image-network:
    driver: bridge