jupyter_gpu/docker-compose.yml
2024-06-02 15:08:03 -04:00

19 lines
544 B
YAML

version: '3.8'
services:
jupyter:
build: /opt/jupyter_gpu
image: cvtt/jupyter_gpu:v1.0.5
container_name: cvtt_gpu_jupyter
runtime: nvidia
environment:
- JUPYTER_ENABLE_LAB=yes
volumes:
- /opt/jupyter_gpu/notebooks:/workspace
- /opt/jupyter_gpu/jupyter_settings/user-settings:/root/.jupyter/lab/user-settings
- /opt/jupyter_gpu/jupyter_settings/workspaces:/root/.jupyter/lab/workspaces
- /opt/jupyter_gpu/.cache/torch:/root/.cache/torch
ports:
- "8888:8888"
shm_size: "8gb"