{
  "id": "nvidia-cuda",
  "version": "3.0.0",
  "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"
    },
    "installCudnnDev": {
      "type": "boolean",
      "default": false,
      "description": "Additionally install CUDA Deep Neural Network (cuDNN) development libraries and headers"
    },
    "installNvtx": {
      "type": "boolean",
      "default": false,
      "description": "Additionally install NVIDIA Tools Extension (NVTX)"
    },
    "installToolkit": {
      "type": "boolean",
      "default": false,
      "description": "Additionally install NVIDIA CUDA Toolkit"
    },
    "cudaVersion": {
      "type": "string",
      "proposals": [
        "12.5",
        "12.4",
        "12.3",
        "12.2",
        "12.1",
        "12.0",
        "11.8",
        "11.7",
        "11.6",
        "11.5",
        "11.4",
        "11.3",
        "11.2"
      ],
      "default": "12.5",
      "description": "Version of CUDA to install"
    },
    "cudnnVersion": {
      "type": "string",
      "proposals": [
        "automatic",
        "8.9.5.29",
        "8.9.4.25",
        "8.9.3.28",
        "8.9.2.26",
        "8.9.1.23",
        "8.9.0.131",
        "8.8.1.3",
        "8.8.0.121",
        "8.7.0.84",
        "8.6.0.163",
        "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",
        "9.0.0.312",
        "9.1.0.70",
        "9.1.1.17",
        "9.2.0.82",
        "9.2.1.18",
        "9.3.0.75",
        "9.4.0.58"
      ],
      "default": "automatic",
      "description": "Version of cuDNN to install"
    }
  },
  "customizations": {
        "vscode": {
            "settings": {
                "github.copilot.chat.codeGeneration.instructions": [
                    {
                        "text": "This dev container includes shared libraries for NVIDIA CUDA pre-installed and available on the `PATH`. It's only useful for dev containers that run on a host machine with an NVIDIA GPU. Within your dev container, use the `nvidia-smi` command to ensure that your GPU is available for CUDA. If the `nvidia-smi` command is not available, you may need to follow NVIDIA's instructions to install the NVIDIA Container Toolkit on your host machine."
                    }
                ]
            }
        }
    },
  "installsAfter": [
      "ghcr.io/devcontainers/features/common-utils"
  ]
}