I install kaggle_python docker image from this tutorial： http://blog.kaggle.com/2016/02/05/how-to-get-started-with-data-science-in-containers/ this image is perfect but I don't know how to use GPU in it. anyone have any idea?
Asked
Active
Viewed 552 times
0
-
Please provide the code you're struggling with in the question. – Kalle Richter Jun 02 '18 at 13:47
2 Answers
0
Nvidia has released a docker runtime that allows docker containers to access their host GPU. Assuming the image you're running has the CUDA libraries built in, you ought to be able to install nvidia-docker as per their instructions, then just launch a container using docker run --runtime=nvidia ...
There's an FAQ for using nvidia-dockers if you run into other roadblocks. I haven't done this myself, but lots of issues are probably going to be specific to how you installed the drivers and cuda libraries on your particular machine. You may also have to modify the image to include any necessary CUDA libraries if they aren't already installed.
scnerd
- 5,836
- 2
- 21
- 36
0
Did you download the CUDA branch (link: https://github.com/Kaggle/docker-python/tree/cuda)? If so, all the infrastructure for the GPUs should already be set up and ready to go. Otherwise, you're going to have to do the setup yourself. :)
Rachael Tatman
- 841
- 7
- 6