31 lines
882 B
YAML
31 lines
882 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}"
|
|
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"
|