Spaces:
Running
on
Zero
Running
on
Zero
File size: 441 Bytes
921d811 fe94868 0f381fd 921d811 fe94868 921d811 cac4e12 921d811 c64341f 921d811 cac4e12 921d811 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import torch
import gradio as gr
import spaces
import nvdiffrast.torch as dr
@spaces.GPU
def pinfo():
print("CUDA Version:", torch.version.cuda)
print(torch.version.cuda)
print(torch.cuda.is_available())
zero = torch.Tensor([0]).cuda()
print(zero.device)
def my_function(input_text):
pinfo()
return f"Received: {input_text}"
iface = gr.Interface(fn=my_function, inputs="text", outputs="text")
iface.launch()
|