Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ class LinearModel(torch.nn.Module):
|
|
19 |
|
20 |
def process(n: int):
|
21 |
print(f'\n===process step {n}')
|
22 |
-
print('cuda visible devices: ' + os.getenv('CUDA_VISIBLE_DEVICES'))
|
23 |
print('cuda avaliable: ' + str(torch.cuda.is_available()))
|
24 |
print('cuda device count: ' + str(torch.cuda.device_count()))
|
25 |
|
|
|
19 |
|
20 |
def process(n: int):
|
21 |
print(f'\n===process step {n}')
|
22 |
+
print('cuda visible devices: ' + str(os.getenv('CUDA_VISIBLE_DEVICES')))
|
23 |
print('cuda avaliable: ' + str(torch.cuda.is_available()))
|
24 |
print('cuda device count: ' + str(torch.cuda.device_count()))
|
25 |
|