27 lines
791 B
YAML
27 lines
791 B
YAML
version: '3.8'
|
|
|
|
#==================================================
|
|
# Relies on the file `.env` content for varables:
|
|
# ROOT_DIR
|
|
# JUPYTER_PORT
|
|
#==================================================
|
|
|
|
services:
|
|
jupyter:
|
|
user: "oleg"
|
|
build: ${ROOT_DIR}
|
|
image: cvtt/jupyter_gpu_tf:v1.0.4
|
|
container_name: jupyter_gpu_tf
|
|
runtime: nvidia
|
|
environment:
|
|
- JUPYTER_ENABLE_LAB=yes
|
|
- 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
|
|
- /opt/jupyter_gpu/data:/workspace/data
|
|
ports:
|
|
- "${JUPYTER_PORT}:${JUPYTER_PORT}"
|
|
shm_size: "8gb"
|