services:
  jupyter_gpu:
    build:
      context: ./jupyter_gpu
      dockerfile: Dockerfile
    image: jupyter_gpu:cuda-12.6-cudnn-runtime
    ports:
      - 0.0.0.0:8000:8000
    volumes:
      - ./notebook:/opt/jupyter/jupyterlab/notebook
    tty: true
    #environment:
    # - NVIDIA_DISABLE_REQUIRE: true
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    profiles:
      - jupyter_gpu
  jupyter:
    build:
      context: ./jupyter
      dockerfile: Dockerfile
    image: jupyter:latest
    ports:
      - 0.0.0.0:8000:8000
    volumes:
      - ./notebook:/opt/jupyter/jupyterlab/notebook
    tty: true
    profiles:
      - jupyter