version: '3.7'
x-heavydbandweb: &heavydbandweb
  image: "heavyai/heavyai-ee-cuda:v6.1.1"
  restart: on-failure
  networks:
    - heavyai-backend
    - heavyai-frontend
services:
  heavydb:
    <<: *heavydbandweb
    deploy:
      resources:
        reservations:
          devices:
            - capabilities:
              - gpu
    container_name: heavydb
    ipc: shareable
    ports:
      - 6274:6274
      - 6276:6276
      - 6278:6278
    environment:
      - CUDA_CACHE_PATH=/var/lib/heavyai/CUDACache
      - CUDA_CACHE_MAXSIZE=4294967296
    volumes:
      - /var/lib/heavyai:/var/lib/heavyai
    ulimits:
      stack: 1073741824
    command: /opt/heavyai/bin/heavydb /var/lib/heavyai/storage --config /var/lib/heavyai/heavy.conf
  heavyweb:
    <<: *heavydbandweb
    container_name: heavyweb
    ports:
      - 6273:6273
    depends_on:
      - heavydb
    volumes:
      - /var/lib/heavyai:/var/lib/heavyai
    command: /opt/heavyai/bin/heavy_web_server --data /var/lib/heavyai/storage --config /var/lib/heavyai/heavy.conf --backend-url http://heavydb:6278
networks:
  heavyai-frontend:
    driver: bridge
    name: heavyai-frontend
  heavyai-backend:
    driver: bridge
    name: heavyai-backend