diff --git a/docker-compose-dev.yml b/docker-compose-dev.yml index 861e5e8..1a38ed0 100644 --- a/docker-compose-dev.yml +++ b/docker-compose-dev.yml @@ -6,6 +6,13 @@ services: image: kamco-cd-training-api:${IMAGE_TAG:-latest} container_name: kamco-cd-training-api runtime: nvidia + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu] ports: - "7200:8080" environment: