From acafefa26838243c4eb70d32dd7e550e59b0a5ac Mon Sep 17 00:00:00 2001
From: artmakh <ar.makhortov@gmail.com>
Date: Tue, 14 Jan 2025 21:09:32 +0700
Subject: [PATCH] fix(ci): Proper GPU forwarding in docker compose

---
 docker-compose-gpu-runner-cuda-12-0.yml | 4 +++-
 docker-compose-gpu-runner.yml           | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/docker-compose-gpu-runner-cuda-12-0.yml b/docker-compose-gpu-runner-cuda-12-0.yml
index 0bc2b6cedb99..dd2311db7bd8 100644
--- a/docker-compose-gpu-runner-cuda-12-0.yml
+++ b/docker-compose-gpu-runner-cuda-12-0.yml
@@ -59,7 +59,9 @@ services:
       resources:
         reservations:
           devices:
-            - capabilities: [ gpu ]
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 
   postgres:
     image: "postgres:14"
diff --git a/docker-compose-gpu-runner.yml b/docker-compose-gpu-runner.yml
index 32665eb7010a..807e567de647 100644
--- a/docker-compose-gpu-runner.yml
+++ b/docker-compose-gpu-runner.yml
@@ -49,7 +49,9 @@ services:
       resources:
         reservations:
           devices:
-            - capabilities: [ gpu ]
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
   postgres:
     image: "postgres:14"
     command: postgres -c 'max_connections=200'