I have built a Docker image from the jupyter/pytorch-notebook.
Now I’d like to get that image to use GPUs when available.
What is the right strategy for doing that?
I don’t know if I can enhance the jupyter stack to add what I need from the Nvidia universe or if I need to start all over again with an Nvidia base and build from there.