Update app.py
Browse files
app.py
CHANGED
@@ -87,9 +87,9 @@ print(zero.device)
|
|
87 |
|
88 |
@spaces.GPU
|
89 |
def test():
|
90 |
-
print('intest', zero.device)
|
91 |
device = zero.device
|
92 |
-
print('1 ',torch.Tensor([1]).
|
93 |
return 0
|
94 |
|
95 |
test()
|
|
|
87 |
|
88 |
@spaces.GPU
|
89 |
def test():
|
90 |
+
print('intest device', zero.device)
|
91 |
device = zero.device
|
92 |
+
print('1 ',torch.Tensor([1]).cuda())
|
93 |
return 0
|
94 |
|
95 |
test()
|