2024-08-02 20:09:53 -04:00

32 lines
902 B
YAML

version: '3.8'
#==================================================
# Relies on the file `.env` content for varables:
# ROOT_DIR
# DATA_DIR
# JUPYTER_PORT
# UID
# GID
#==================================================
services:
jupyter:
# user: "${UID}:${GID}"
# build: ${ROOT_DIR}
image: cloud21.cvtt.vpn:5500/jupyter_gpu_pt: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"