Nvidia DGX-2 - 4.6.2 Using the NVIDIA Container Runtime for Docker

nvidia-docker2)

$ docker run --runtime=nvidia ...

$ nvidia-docker run ...
nvidia-docker2
nvidia-docker
 docker run nvidia
nvidia
/etc/docker/daemon.json
"default-runtime": "nvidia",
{
"default-runtime": "nvidia",
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
},
}
docker run
$ docker run ...
!
CAUTION: If you build Docker images while nvidia is set as the default
runtime, make sure the build scripts executed by the Dockerfile specify the GPU
architectures that the container will need. Failure to do so may result in the
container being optimized only for the GPU architecture on which it was built.