Join GitHub today
GitHub is home to over 20 million developers working together to host and review code, manage projects, and build software together.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
Already on GitHub? Sign in to your account
support for running containers on nodes with GPU #194
Comments
malchun
self-assigned this
May 11, 2017
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
anaderi commentedMar 31, 2017
to run code inside container that runs something on nvidia GPU, one has to take care of certain options for docker that tell it to bind device and drivers inside container (of course image it run should be created accordingly -- TODO: describe on wiki).
there is a helper project that takes care of everything: https://github.com/NVIDIA/nvidia-docker, it creates a special service (listens to http://localhost:3476 and tells additional arguments to docker: https://github.com/NVIDIA/nvidia-docker/wiki/nvidia-docker-plugin, i.e. http://localhost:3476/docker/cli/json returns json with all additional args that would enable running GPU-specific stuff inside container, i.e.
nvidia-smi)in particular there are
devicesarguments that docker-py of version 2.0+ can understand.So the task would be to
docker.client.run