Spaces:
Runtime error
Runtime error
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 |
################################################################
|