Support for GPUs (NVIDIA CUDA)

[cgddrd]

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]

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

[abng88]

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]

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