Update devcontainer.json

Cudnn and extensions
wandb · Sep 27, 2023 · 2d820c6 · 2d820c6
1 parent e1d9da1
commit 2d820c6
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -1,7 +1,17 @@
 {
   "image": "mcr.microsoft.com/devcontainers/universal:2",
+  "customizations": {
+    "vscode": {
+      "extensions": [
+        "ms-python.python",
+        "ms-toolsai.jupyter"
+      ]
+    }
+  },
   "features": {
-    "ghcr.io/devcontainers/features/nvidia-cuda:1": {},
+    "ghcr.io/devcontainers/features/nvidia-cuda:1": {
+      "installCudnn": true
+    },
     "ghcr.io/devcontainers/features/docker-outside-of-docker:1": {}
   }
 }