From 80365b9dc664ddf6178f16d7aad451797cca39c4 Mon Sep 17 00:00:00 2001
From: Tobit Flatscher <53856473+2b-t@users.noreply.github.com>
Date: Sun, 9 Jul 2023 22:32:15 +0100
Subject: [PATCH] feat: Re-add Docker-Compose file for Nvidia GPUs

---
 .devcontainer/devcontainer.json      |  2 +-
 docker/docker-compose-gui-nvidia.yml | 10 ++++++++++
 2 files changed, 11 insertions(+), 1 deletion(-)
 create mode 100644 docker/docker-compose-gui-nvidia.yml

diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 6e67b86..4ee3fc2 100644
--- 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
new file mode 100644
index 0000000..615c731
--- /dev/null
+++ 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