{
"id": "nvidia-cuda",
"version": "1.0.3",
"name": "NVIDIA CUDA",
"description": "Installs shared libraries for NVIDIA CUDA.",
"documentationURL": "https://github.com/devcontainers/features/tree/main/src/nvidia-cuda",
"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"
}
}
}
