diff --git a/docker-compose.yaml b/docker-compose.yaml index 01f0f7c45cee81accceaf3579c60e6f0deb7e9ed..0e6308a2b2148545ae916f42394ce515fea86563 100755 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -138,7 +138,9 @@ services: resources: reservations: devices: - - capabilities: [gpu] + - driver: nvidia + device_ids: ['1'] + capabilities: [gpu] cpus: '10' memory: 30G ipc: host