Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
•
88d43bf
1
Parent(s):
693a19d
Look for cuda
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ from huggingface_hub import hf_hub_download
|
|
| 29 |
|
| 30 |
def main():
|
| 31 |
|
| 32 |
-
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
device = "cuda:0"
|
|
|
|
| 29 |
|
| 30 |
def main():
|
| 31 |
|
| 32 |
+
os.system("find / -name nvcc")
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
device = "cuda:0"
|