Support for GPUs (NVIDIA CUDA)

manifoldai / docker-cookiecutter-data-science

A fork of the cookiecutter-data-science leveraging Docker for local development.

http://drivendata.github.io/cookiecutter-data-science/

MIT License

130 stars 29 forks source link

Support for GPUs (NVIDIA CUDA) #1

Open cgddrd opened 6 years ago

cgddrd commented 6 years ago

Hi All,

This project looks very interesting - is there any intention of supporting GPUs as part the ML base docker image/Cookiecutter project (e.g. via NVIDIA CUDA)?

I'm happy to look into this myself, but thought it was worth checking if this was on the horizon in terms of this project?

Thanks,

sergeyk commented 6 years ago

I love the idea and how you've made it actionable! Supporting GPUs is crucial to actually using this for deep learning development :)

abng88 commented 5 years ago

Hey @cgddrd and @sergeyk we have some new base images for out of the box GPU support. Will be putting them up very shortly!

cgddrd commented 5 years ago

@abng88 - This is awesome, looking forward to getting stuck in!