ml-tooling / ml-hub

🧰 Multi-user development platform for machine learning teams. Simple to setup within minutes.
Apache License 2.0
301 stars 64 forks source link

Docker 19.03 GPU support #12

Open SpontaneousDuck opened 4 years ago

SpontaneousDuck commented 4 years ago

Describe the issue: Docker 19.03 and nvidia-docker changed how gpus are bound to a container. Instead of calling the runtime, all you need to do is pass in --gpus 2 to docker run. Would it be possible to support this? My systems are running v19.03. Thanks!

Technical details: