Docker compose has nice support for GPUs, K8s has moved their cluster-wide GPU scheduler from experimental to stable status. Docker swarm has yet to support the device
option used in docker compose
so the mechanisms for supporting GPUs on swarm are a bit more open-ended.
- NVIDIA container runtime for docker. The runtime is no longer required to run GPU support with the docker cli or compose; however, it appears necessary so that one can set
Default Runtime: nvidia
for swarm mode. - docker compose GPU support
- Good GitHub Gist Reference for an overview on Swarm with GPUs. It is a bit dated, but has good links and conversation.
- [Miscellaneous Opt