ml-hub
ml-hub copied to clipboard
Docker 19.03 GPU support
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:
- Hub version : 1.0.0
- Docker version : 19.03.10
- Host Machine OS (Windows/Linux/Mac): Linux
- Command used to start the hub :
sudo docker run --rm -p 8095:8080 -v /var/run/docker.sock:/var/run/docker.sock mltooling/ml-hub:latest
- Browser (Chrome/Firefox/Safari): Chromium