add gpu argument to devcontainer gpu configuration

testingautomated-usi · Aug 14, 2024 · 0d42321 · 0d42321
1 parent 33ba735
commit 0d42321
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.devcontainer/gpu/devcontainer.json b/.devcontainer/gpu/devcontainer.json
@@ -1,4 +1,5 @@
 {
     "name": "muprl - GPU",
-    "image": "dockercontainervm/muprl-gpu:0.1.0"
+    "image": "dockercontainervm/muprl-gpu:0.1.0",
+    "runArgs": ["--gpus", "all"]
 }