Updated devcontainer

airtai · Jan 14, 2025 · 08f5e7f · 08f5e7f
1 parent 108db2d
commit 08f5e7f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.devcontainer/cuda-tf/devcontainer.json b/.devcontainer/cuda-tf/devcontainer.json
@@ -37,8 +37,9 @@
         "ghcr.io/iterative/features/nvtop:1": {}
     },
     "updateContentCommand": "bash .devcontainer/cuda-tf/setup.sh",
-    //     "nvidia-smi"
-    // ],
+    "postCreateCommand": [
+        "nvidia-smi"
+    ],
     "customizations": {
         "vscode": {
             "settings": {