Update app.py
Browse files
app.py
CHANGED
@@ -20,9 +20,9 @@ print('outside',zero.device)
|
|
20 |
@spaces.GPU
|
21 |
def test(n):
|
22 |
print(n)
|
23 |
-
|
24 |
# device = zero.device
|
25 |
-
|
26 |
return 0
|
27 |
|
28 |
test(1)
|
|
|
20 |
@spaces.GPU
|
21 |
def test(n):
|
22 |
print(n)
|
23 |
+
print('insidew device', zero.device)
|
24 |
# device = zero.device
|
25 |
+
print('1 ',torch.Tensor([1]).cuda())
|
26 |
return 0
|
27 |
|
28 |
test(1)
|