Skip to content

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.

Edited by Damien Gerard