jupyter_gpu/tensorflow/docker-compose.yml
2024-08-07 21:01:30 +00:00

28 lines
839 B
YAML

version: '3.8'
#==================================================
# Relies on the file `.env` content for varables:
# ROOT_DIR
# DATA_DIR
# JUPYTER_PORT
#==================================================
services:
jupyter:
image: cloud21.cvtt.vpn:5500/jupyter_gpu_tensorflow:latest
container_name: jupyter_gpu_tf
runtime: nvidia
restart: unless-stopped
environment:
- JUPYTER_ENABLE_LAB=yes
#- PYTHONPATH=/cvtt/prod
- TF_CACHE_DIR=/workspace/tf_cache
volumes:
- ${ROOT_DIR}/notebooks:/workspace
- ${ROOT_DIR}/jupyter_settings/user-settings:/home/oleg/.jupyter/lab/user-settings
- ${ROOT_DIR}/jupyter_settings/workspaces:/home/oleg/.jupyter/lab/workspaces
- ${DATA_DIR}:/workspace/data
ports:
- "${JUPYTER_PORT}:${JUPYTER_PORT}"
shm_size: "8gb"