Add new NVIDIA CUDA feature (#80)

* Add new Nvidia feature

* Remove random empty files

* Update comments

* Rename feature to nvidia-cuda

* Add feature to tests

* Add version

* Move test to match new name

* Add final output message

* Fix capitalization of NVIDIA

* Remove option for base CUDA

* Use camelCase

* Check for required packages

* Use os-release instead of lsb_release

* Clean up keyring variables

* Collapse keyring lines

* Always install CUDA libraries

* Add option to install NVTX

* Always use ubuntu2004 repo

* Use test instead of brackets

* Add default values to feature

* Add version options for CUDA and cuDNN

* Rename CUDA version option

* Add scenario to test specific CUDA/cuDNN version

* Rename cuDNN scenario

* Fix typo in test scenario

* Update variable casing

* Add more helpful error messages

* Remove default values from script

* Use enum for version option

* Polish new scenarios

* Remove apt_get_update_if_needed and check_packages

* Add more versions

* Improve error messages

* Comments and feature description

Co-authored-by: Josh Spicer <joshspicer@github.com>

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"
+    }
+  }
+}