Add Docker environment setup
This commit is contained in:
21
docker-compose.yml
Normal file
21
docker-compose.yml
Normal file
@@ -0,0 +1,21 @@
|
||||
services:
|
||||
jupyter:
|
||||
build: .
|
||||
container_name: gie-eris-jupyter
|
||||
ports:
|
||||
- "${JUPYTER_PORT:-8888}:8888"
|
||||
volumes:
|
||||
- ./notebooks:/workspace/notebooks
|
||||
- ./media-pipeline:/workspace/media-pipeline
|
||||
- ./scripts:/workspace/scripts
|
||||
- ${HF_CACHE_DIR:-./.cache/huggingface}:/workspace/.cache/huggingface
|
||||
env_file:
|
||||
- .env
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
count: all
|
||||
capabilities: [gpu]
|
||||
restart: unless-stopped
|
||||
Reference in New Issue
Block a user