From 2e928d96edd3d8e3f154ba23d07a4e4e797bf421 Mon Sep 17 00:00:00 2001 From: JP Ungaretti <19893438+jungaretti@users.noreply.github.com> Date: Mon, 29 Aug 2022 16:18:02 -0500 Subject: [PATCH] Add NOTES.md for CUDA feature (#119) --- src/nvidia-cuda/NOTES.md | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 src/nvidia-cuda/NOTES.md diff --git a/src/nvidia-cuda/NOTES.md b/src/nvidia-cuda/NOTES.md new file mode 100644 index 0000000..b19fde2 --- /dev/null +++ b/src/nvidia-cuda/NOTES.md @@ -0,0 +1,19 @@ +## Compatibility + +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: + +### Install the NVIDIA Container Toolkit + +Follow [NVIDIA's instructions to install the NVIDIA Container Toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/overview.html) 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 + +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: + +```json +{ + "runArgs": ["--gpus", "all"] +} +```