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