immich/docker/docker-compose.gpu.yml

version: "3.8"

services:
  immich_server:
    image: immich-server-dev:1.3.2
    build:
      context: ../server
      target: development
      dockerfile: ../server/Dockerfile
    command: npm run start:dev
    expose:
      - "3000"
    volumes:
      - ../server:/usr/src/app
      - ${UPLOAD_LOCATION}:/usr/src/app/upload
      - /usr/src/app/node_modules
    env_file:
      - .env
    depends_on:
      - redis
      - database
    networks:
      - immich_network

  redis:
    container_name: immich_redis
    image: redis:6.2
    networks:
      - immich_network

  database:
    container_name: immich_postgres
    image: postgres:14
    env_file:
      - .env
    environment:
      POSTGRES_PASSWORD: ${DB_PASSWORD}
      POSTGRES_USER: ${DB_USERNAME}
      POSTGRES_DB: ${DB_DATABASE_NAME}
      PG_DATA: /var/lib/postgresql/data
    volumes:
      - pgdata:/var/lib/postgresql/data
    ports:
      - 5432:5432
    networks:
      - immich_network

  nginx:
    container_name: proxy_nginx
    image: nginx:latest
    volumes:
      - ./settings/nginx-conf:/etc/nginx/conf.d
    ports:
      - 2283:80
      - 2284:443
    logging:
      driver: none
    networks:
      - immich_network
    depends_on:
      - immich_server

  immich_tf_fastapi:
    container_name: immich_tf_fastapi
    image: tensor_flow_fastapi:1.0.0
    restart: always
    command: uvicorn app.main:app --proxy-headers --host 0.0.0.0 --port 8000 --reload
    build:
      context: ../machine_learning
      target: gpu
      dockerfile: ../machine_learning/Dockerfile
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    volumes:
      - ../machine_learning/app:/code/app
      - ${UPLOAD_LOCATION}:/code/app/upload
    ports:
      - 2285:8000
    expose:
      - "8000"
    depends_on:
      - database
    networks:
      - immich_network

      
networks:
  immich_network:
volumes:
  pgdata: