Support for nvidia-docker
Description
gitlab-ci already supports Docker container, but the normal Docker containers do not have GPU capabilities. Docker images with GPU capabilities are used by researchers for Deep learning and other IA related projects.
Proposal
One common solution is to use nvidia-docker
, which runs on top of Docker. Basically
it is like running docker, except you use the binary nvidia-docker
instead of
the traditional docker
. nvidia-docker
may have additional & specific parameters
but this is not required in a first time.
A simple implementation in Gitlab CI would probably be to use the standard docker
executor whenever
the executor docker-nvidia
is selected for an agent.
Another strategy would be to be able to change the binary path from the configuration file.
Links to related issues and merge requests / references
Question
I may propose a simple MR. But I would appreciate guidance for the best strategy to take from the Core team.