Update devcontainer.json

SoCSTech · Sep 6, 2024 · 46bc272 · 46bc272
1 parent 6948a95
commit 46bc272
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/CMP9137 Advanced Machine Learning/.devcontainer/devcontainer.json b/CMP9137 Advanced Machine Learning/.devcontainer/devcontainer.json
@@ -2,6 +2,7 @@
     "name": "Advanced Machine Learning",
 
     "image": "ghcr.io/socstech/cmp9137:latest",
+    "runArgs": ["--gpus=all"],
 
     //"build": { "dockerfile": "Dockerfile" },
 
@@ -20,4 +21,4 @@
             ]
         }
     }
-}
+}