Read GPU Id from .env file.

wilhelm-lab · Aug 28, 2023 · c63e9d1 · c63e9d1
1 parent 621d36f
commit c63e9d1
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/.env b/.env
@@ -1,3 +1,4 @@
 GID=1000
 UID=1000
 COMPOSE_PROJECT_NAME=koina_1000
+GPU_DEVICE=0
diff --git a/.env.ci b/.env.ci
@@ -1,3 +1,4 @@
 GID=1001
 UID=1001
 COMPOSE_PROJECT_NAME=koina_ci
+GPU_DEVICE=3
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -11,7 +11,7 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              device_ids: ['0']
+              device_ids: ["${GPU_DEVICE}"]
               capabilities: [ gpu ]
     shm_size: 2GB
     restart: unless-stopped