hysts-debug/debuggingapp

1import os2 3os.environ["NUMBA_DISABLE_CUDA"] = "1"4 5import gradio as gr6import spaces7import torch8 9 10zero = torch.Tensor([0]).cuda()11print(zero.device)  # <-- 'cpu' 🤔12 13 14@spaces.GPU15def greet(n):16    print(zero.device)  # <-- 'cuda:0' 🤗17    return f"Hello {zero + n} Tensor"18 19 20demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())21demo.launch()22