A minimal example demonstrating CUDA in docker-based buildkite pipelines using a two step pipeline:
-
A
nvidia-cuda
&conda
based "test runner" is built, including both os-level and conda-level requirements. This runs within the buildkit-agent container. -
The
docker-buildkite-plugin
is used to execute tests via thenvidia
docker runtime within the runner container.
The example requires an agent host with
nvidia-docker installed. The
buildkite agent may be run via the standard
buildkite/agent
image,
bind-mounting /var/run/docker.sock
to enable support for
docker-buildkite-plugin
.