Fix engine NVIDIA GPU guide

Our guide on setting up the engine to use NVIDIA GPUs refers to
deprecated resources. nvidia-container-runtime has been superseded by
nvidia-container-toolkit. So update our docs to reflect that.

Signed-off-by: Piotr Stankiewicz <piotr.stankiewicz@docker.com>
This commit is contained in:
Piotr Stankiewicz 2024-02-09 15:28:44 +01:00
parent 27645332ad
commit d4f00293ab
1 changed files with 7 additions and 20 deletions

View File

@ -273,22 +273,9 @@ done so.
Verify that your GPU is running and accessible.
#### Install nvidia-container-runtime
#### Install nvidia-container-toolkit
Follow the instructions at (https://nvidia.github.io/nvidia-container-runtime/)
and then run this command:
```console
$ apt-get install nvidia-container-runtime
```
Ensure the `nvidia-container-runtime-hook` is accessible from `$PATH`.
```console
$ which nvidia-container-runtime-hook
```
Restart the Docker daemon.
Follow the official NVIDIA Container Toolkit [installation instructions](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html).
#### Expose GPUs for use
@ -350,10 +337,10 @@ This enables the `utility` driver capability which adds the `nvidia-smi` tool to
the container.
Capabilities as well as other configurations can be set in images via
environment variables. More information on valid variables can be found at the
[nvidia-container-runtime](https://github.com/NVIDIA/nvidia-container-runtime)
GitHub page. These variables can be set in a Dockerfile.
environment variables. More information on valid variables can be found in the
[nvidia-container-toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/docker-specialized.html)
documentation. These variables can be set in a Dockerfile.
You can also use CUDA images which sets these variables automatically. See the
[CUDA images](https://github.com/NVIDIA/nvidia-docker/wiki/CUDA) GitHub page
for more information.
official [CUDA images](https://catalog.ngc.nvidia.com/orgs/nvidia/containers/cuda)
NGC catalog page.