17 lines
426 B
YAML
17 lines
426 B
YAML
version: '3.8'
|
|
|
|
services:
|
|
vr180-matting:
|
|
build: .
|
|
runtime: nvidia
|
|
environment:
|
|
- NVIDIA_VISIBLE_DEVICES=all
|
|
- CUDA_VISIBLE_DEVICES=0
|
|
volumes:
|
|
- ./data:/app/data # Mount data directory
|
|
- ./output:/app/output # Mount output directory
|
|
- ./models:/app/models # Mount models directory
|
|
working_dir: /app
|
|
stdin_open: true
|
|
tty: true
|
|
command: /bin/bash |