$ docker run --runtime=nvidia ...
$ nvidia-docker run ...
nvidia-docker2
nvidia-docker
docker run
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.
nvidia
nvidia-docker2)