table-to-csv-pipeline / docker-compose.yml
rizgiak's picture
initial commit
4df4988
raw
history blame contribute delete
451 Bytes
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 ]