File size: 451 Bytes
4c41a36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
version: "3.7"
services:
model_name:
build:
context: .
dockerfile: .docker/Dockerfile
container_name: model_name
ports:
- "8996:8996"
env_file:
- ./.env
volumes:
- ./data:/home/working/data:ro
# This part is used to enable GPU support
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [ gpu ]
|