hiascend commited on
Commit
6e29d7d
·
verified ·
1 Parent(s): 563fb2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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