Update app.py

app.py

@@ -1,10 +1,14 @@
 import gradio as gr
-import spaces
+import spaces  # 导入spaces时会应用torch补丁
 import torch
 
+spaces.zero.torch.unpatch()  # 手动调用解除torch补丁方法
+
+print('cuda avaliable: ' + str(torch.cuda.is_available()))  # false
+
 torch.set_default_device('cuda')
 
-zero = torch.Tensor([0]).cuda()
+zero = torch.Tensor([0]).cuda()  # 报错：无GPU资源
 
 one = torch.ones(2, 2, device='cuda')
 
@@ -13,7 +17,6 @@ two = torch.matmul(one, one).cuda()
 
 def print_device():
     print('cuda avaliable: ' + str(torch.cuda.is_available()))
-    # print('default device: ' + str(torch.get_default_device()))  # 报错：AttributeError: module 'torch' has no attribute 'get_default_device'
     print('zero device: ' + str(zero.device))
     print('one device: ' + str(one.device))
     print('two device: ' + str(two.device))