diff --git a/docker-compose.yml b/docker-compose.yml index b224e1ac3..1f8dc9f52 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -50,7 +50,7 @@ services: reservations: devices: - driver: nvidia - device_ids: ['1'] + device_ids: ['0'] capabilities: [gpu]