From 5641795d81553b029a24c42d1ca1eeaf7d7c4384 Mon Sep 17 00:00:00 2001 From: joecryptotoo <80373433+joecryptotoo@users.noreply.github.com> Date: Thu, 22 Jun 2023 16:23:02 -0700 Subject: [PATCH] changed to gpu 0(first GPU) For me GPU0 is a 3080 and GPU1 is a 4090 others may only have one GPU --- docker-compose.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]