Spaces:
Running
on
L40S
Running
on
L40S
Upload app_hg.py with huggingface_hub
Browse files
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 |
################################################################
|