I have a GPU application that does unit-testing during the image building stage. With Docker 19.03, one can specify nvidia runtime with
docker run --gpus all but I also need access to the gpus for
docker build because I do unit-testing. How can I achieve this goal?
For older version of docker that use nvidia-docker2 it was not possible to specifiy runtime during build stage, BUT you can set the default runtime to be nvidia, and docker build works fine that way. Can I do that in Docker 19.03 that doesn’t need nvidia-docker anymore? If so, how?