version: '3.8'

#==================================================
# Relies on the file `.env` content for varables:
#    ROOT_DIR
#    DATA_DIR
#    JUPYTER_PORT
#    UID
#    GID
#==================================================
services:
  jupyter:
    # user: "${UID}:${GID}"
    image: cloud21.cvtt.vpn:5500/jupyter_gpu_pytorch:latest
    container_name: jupyter_gpu_pt
    runtime: nvidia
    restart: unless-stopped
    environment:
      - JUPYTER_ENABLE_LAB=yes
      - PYTHONPATH=/cvtt/prod
    volumes:
      - ${ROOT_DIR}/notebooks:/workspace
      - ${ROOT_DIR}/jupyter_settings/user-settings:/root/.jupyter/lab/user-settings
      - ${ROOT_DIR}/jupyter_settings/workspaces:/root/.jupyter/lab/workspaces
      - ${ROOT_DIR}/.cache/torch:/root/.cache/torch

      - ${DATA_DIR}:/workspace/data
    ports:
      - "${JUPYTER_PORT}:${JUPYTER_PORT}"
    shm_size: "8gb"