dongyh20 commited on
Commit
7b80a5d
·
1 Parent(s): a5173a0

update space

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  os.system('nvidia-smi')
3
  print(torch.backends.cudnn.version())
4
  os.environ['LOWRES_RESIZE'] = '384x32'
 
1
  import os
2
+ import torch
3
  os.system('nvidia-smi')
4
  print(torch.backends.cudnn.version())
5
  os.environ['LOWRES_RESIZE'] = '384x32'