Huiwenshi commited on
Commit
7cfde4b
1 Parent(s): 1c17889

Upload app_hg.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app_hg.py +3 -0
app_hg.py CHANGED
@@ -97,6 +97,9 @@ except Exception as err:
97
 
98
  from glob import glob
99
  print(glob("/usr/local/cuda/*"))
 
 
 
100
  ################################################################
101
  # initial setting
102
  ################################################################
 
97
 
98
  from glob import glob
99
  print(glob("/usr/local/cuda/*"))
100
+ print(torch.cuda.is_available())
101
+ print(os.environ.get('CUDA_HOME', None))
102
+ os.environ['CUDA_HOME'] = '/usr/local/cuda'
103
  ################################################################
104
  # initial setting
105
  ################################################################