diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 3514d8d0..41db75f8 100644
--- 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": {}
   }
 }