feat: Re-add Docker-Compose file for Nvidia GPUs

2b-t · Jul 9, 2023 · 80365b9 · 80365b9
1 parent ac2e415
commit 80365b9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -1,7 +1,7 @@
 {
   "name": "Docker PREEMPT_RT real-time",
   "dockerComposeFile": [
-    "../docker/docker-compose.yml" // Alternative: "../docker/docker-compose-gui.yml"
+    "../docker/docker-compose.yml" // Alternative: "../docker/docker-compose-gui.yml", "../docker/docker-compose-gui-nvidia.yml"
   ],
   "service": "docker_realtime",
   "workspaceFolder": "/benchmark",

diff --git a/docker/docker-compose-gui-nvidia.yml b/docker/docker-compose-gui-nvidia.yml
@@ -0,0 +1,10 @@
+version: "3.9"
+services:
+  docker_realtime:
+    extends:
+      file: docker-compose-gui.yml
+      service: docker_realtime
+    environment:
+     - NVIDIA_DRIVER_CAPABILITIES=all
+     - NVIDIA_VISIBLE_DEVICES=all
+    runtime: nvidia