compose.yaml

services:
  nerfstudio-renderer:
    image: j3soon/nerfstudio-renderer
    build:
      context: nerfstudio_renderer
      args:
        - CUDA_VERSION=11.8.0
        - CUDA_ARCHITECTURES=86
        - OS_VERSION=22.04
        - SERVER_PORT=10001
    container_name: nerfstudio-renderer
    ports:
      - "10001:10001"
    environment:
      - DISPLAY=$DISPLAY
    volumes:
      - /tmp/.X11-unix:/tmp/.X11-unix
      - ./nerfstudio_renderer/src:/src:ro
      - ./pygame_viewer:/pygame_viewer:ro
      - ./assets:/workspace:ro
      - cache:/home/user/.cache
    shm_size: '6gb'
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
  pygame-viewer:
    image: j3soon/pygame-viewer
    build:
      context: pygame_viewer
    container_name: pygame-viewer
    stdin_open: true
    tty: true
    network_mode: host
    environment:
      - DISPLAY=$DISPLAY
    volumes:
      - /tmp/.X11-unix:/tmp/.X11-unix
      - ./pygame_viewer:/src:ro
  isaac-sim-viewer:
    # Ref: https://github.com/j3soon/isaac-extended?tab=readme-ov-file#docker-container-with-display
    image: j3soon/isaac-sim-viewer
    build:
      context: extension
    container_name: isaac-sim-viewer
    entrypoint: [bash]
    stdin_open: true
    tty: true
    network_mode: host
    environment:
      - ACCEPT_EULA=Y
      - PRIVACY_CONSENT=Y
      - DISPLAY=$DISPLAY
    volumes:
      - ~/docker/isaac-sim/cache/kit:/isaac-sim/kit/cache:rw
      - ~/docker/isaac-sim/cache/ov:/root/.cache/ov:rw
      - ~/docker/isaac-sim/cache/pip:/root/.cache/pip:rw
      - ~/docker/isaac-sim/cache/glcache:/root/.cache/nvidia/GLCache:rw
      - ~/docker/isaac-sim/cache/computecache:/root/.nv/ComputeCache:rw
      - ~/docker/isaac-sim/logs:/root/.nvidia-omniverse/logs:rw
      - ~/docker/isaac-sim/data:/root/.local/share/ov/data:rw
      - ~/docker/isaac-sim/documents:/root/Documents:rw
      - /tmp/.X11-unix:/tmp/.X11-unix
      - ./assets:/workspace
      - ./extension:/src
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
volumes:
  cache: