Need Example Docker File with GPU Support

Spaces
1

Hi,
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”.
Thanks.

2

i think @chris-rannou can help!

3

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

4

A couple of templates examples for you

5

I appreciate it. I’ll check them out.

1 Like