docker-compose.yml

version: "3.9"

# docker compose run --service-ports --name netspresso-dev netspresso bash

services:
  netspresso:
    build:
      context: .
      dockerfile: Dockerfile
    image: netspresso:${TAG}
    container_name: netspresso
    ipc: host
    ports:
      # HOST_PORT for tensorboard
      - 50001:50001 # tensorboard
      - 50002:50002 # jupyter notebook
      - 50003:50003 # (optional, gradio) configuration helper
      - 50004:50004 # (optional, gradio) inference demo
    volumes:
      # from path: your working directory
      # - /PATH/TO/netspresso:/home/appuser/netspresso
      # from path: your dataset directory
      - /PATH/TO/DATA:/DATA/PATH/IN/CONTAINER
      # from path: your checkpoint directory
      - /PATH/TO/CHECKPOINT:/CHECKPOINT/PATH/IN/CONTAINER
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["0", "1", "2", "3"] # your GPU id(s)
              capabilities: [gpu]