Spaces:

hiascend
/

jyh-zero

Runtime error

hiascend commited on Apr 24, 2024

Commit

668888e

verified ·

1 Parent(s): 5326ed4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ 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()))
     print('zero device: ' + str(zero.device))
     print('one device: ' + str(one.device))
     print('two device: ' + str(two.device))

 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))