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