Need Example Docker File with GPU Support

Are there good examples of docker files which run apps using a GPU? I’ve copied the docker file statements from a log for a Gradio space with a GPU enabled to create my docker file. But the Python statement “torch.cuda.is_available()” yields “false”.

i think @chris-rannou can help!

We dont expose GPU at build time, tha’ts this instruction will never work at build time, but should work at runtime

A couple of templates examples for you

I appreciate it. I’ll check them out.

1 Like