Installs shared libraries for NVIDIA CUDA.
"features": {
"ghcr.io/devcontainers/features/nvidia-cuda:1": {}
}| Options Id | Description | Type | Default Value |
|---|---|---|---|
| installCudnn | Additionally install CUDA Deep Neural Network (cuDNN) shared library | boolean | false |
| installNvtx | Additionally install NVIDIA Tools Extension (NVTX) | boolean | false |
| cudaVersion | Version of CUDA to install | string | 11.8 |
| cudnnVersion | Version of cuDNN to install | string | 8.6.0.163 |
This Feature adds shared libraries for NVIDIA CUDA and is only useful for devcontainers that run on a host machine with an NVIDIA GPU. Within your devcontainer, use the nvidia-smi command to ensure that your GPU is available for CUDA.
If the nvidia-smi command is not available within your devcontainer, you may need to complete the following steps:
Follow NVIDIA's instructions to install the NVIDIA Container Toolkit on your host machine. The NVIDIA Container Toolkit is available on a variety of Linux distributions. Make sure you have installed the NVIDIA driver for your Linux distribution before installing the NVIDIA Container Toolkit.
Enable GPU passthrough to your devcontainer by adding ["--gpus", "all"] to your devcontainer's runArgs property. Here's an example of a devcontainer with this property:
{
"runArgs": ["--gpus", "all"]
}This Feature should work on recent versions of Debian/Ubuntu-based distributions with the apt package manager installed.
bash is required to execute the install.sh script.
Note: This file was auto-generated from the devcontainer-feature.json. Add additional notes to a NOTES.md.