1 files changed, 51 insertions, 0 deletions
diff --git a/src/nvidia-cuda/devcontainer-feature.json b/src/nvidia-cuda/devcontainer-feature.json
new file mode 100644
index 0000000..711374b
--- /dev/null
+++ b/src/nvidia-cuda/devcontainer-feature.json
@@ -0,0 +1,51 @@
+{
+  "id": "nvidia-cuda",
+  "name": "NVIDIA CUDA",
+  "description": "Installs shared libraries for NVIDIA CUDA.",
+  "version": "1.0.0",
+  "options": {
+    "installCudnn": {
+      "type": "boolean",
+      "default": false,
+      "description": "Additionally install CUDA Deep Neural Network (cuDNN) shared library"
+    },
+    "installNvtx": {
+      "type": "boolean",
+      "default": false,
+      "description": "Additionally install NVIDIA Tools Extension (NVTX)"
+    },
+    "cudaVersion": {
+      "type": "string",
+      "enum": [
+        "11.7",
+        "11.6",
+        "11.5",
+        "11.4",
+        "11.3",
+        "11.2"
+      ],
+      "default": "11.7",
+      "description": "Version of CUDA to install"
+    },
+    "cudnnVersion": {
+      "type": "string",
+      "enum": [
+        "8.5.0.96",
+        "8.4.1.50",
+        "8.4.0.27",
+        "8.3.3.40",
+        "8.3.2.44",
+        "8.3.1.22",
+        "8.3.0.98",
+        "8.2.4.15",
+        "8.2.2.26",
+        "8.2.1.32",
+        "8.2.0.53",
+        "8.1.1.33",
+        "8.1.0.77"
+      ],
+      "default": "8.5.0.96",
+      "description": "Version of cuDNN to install"
+    }
+  }
+}