Skip to content

nvidia-docker bootstrapping
Compare
Choose a tag to compare
@tomdaley92 tomdaley92 released this 26 Dec 21:39
· 285 commits to stable since this release
2.10.0
dea7e8e
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

Add support for bootstrapping special docker swarm hosts that have access to Nvidia GPUs.

Installs and configures:

  • Nvidia drivers for Ubuntu
  • nvidia-docker-toolkit via nvidia-docker2
  • sets nvidia as default docker runtime

To tell a swarm service to use the GPUs (in a docker-compose.yaml file) you must pass the following environment variables:

  • NVIDIA_DRIVER_CAPABILITIES=compute,video,utility
  • NVIDIA_VISIBLE_DEVICES=all

You can read more about it on the official documentation here

Assets 2
Loading