Re:Show HN: Dockerized GPU Deep Learning Solution (Code and ...
by __mp on Friday January 08, 2016 @08:35AM
(#10864690)
Does docker support gpus or how does this work out?
[ Parent
| on HN ]
Re:Show HN: Dockerized GPU Deep Learning Solution (Code and ...
by viklas on Friday January 08, 2016 @09:02AM
(#10864847)
The "--device" flag allows you to map devices through to a Docker container. It runs in 'privileged' mode though, so isn't suitable for a shared host.
Nvidia make it pretty straight-forward now but we had to branch from that approach a bit for the CoreOS deployment.
https://github.com/NVIDIA/nvidia-docker (Nice pictures)
https://docs.docker.com/engine/reference/run/ (Docker documentation, search for 'privileged')
The approach is a bit different depending on your host operating system. You'll also find there are constraints when you introduce a virtualisation layer, like virtualbox or parallels on your desktop - GPUs can be mapped through, but it's painful(ish).
Re:Show HN: Dockerized GPU Deep Learning Solution (Code and ...
by m-i-l on Friday January 08, 2016 @09:03AM
(#10864852)
There's also a blog entry about accessing GPU from Docker at http://marconijr.com/posts/docker-exposing-gpu/ .