Files
face/docker-compose.yml
2024-07-09 02:29:20 +00:00

21 lines
492 B
YAML

services:
tritonserver:
container_name: tritonserver
image: nvcr.io/nvidia/tritonserver:24.05-py3
command: tritonserver --model-repository=/models
ports:
- 8000:8000
- 8001:8001
- 8002:8002
volumes:
- ${LOCAL_WORKSPACE_FOLDER}/model_repository:/models
privileged: true
deploy:
resources:
reservations:
devices:
- capabilities: [gpu]
networks:
default:
name: face_devcontainer_default