aboutsummaryrefslogtreecommitdiff
path: root/src/nvidia-cuda/devcontainer-feature.json
blob: 76a5ec279944910c7c0fe622244d48f6468e84c2 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
{
  "id": "nvidia-cuda",
  "version": "1.0.4",
  "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"
    }
  },
  "installsAfter": [
      "ghcr.io/devcontainers/features/common-utils"
  ]
}