Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
•
437d430
1
Parent(s):
88d43bf
find cuda
Browse files- app.py +1 -1
- pre-requirements.txt +2 -2
app.py
CHANGED
|
@@ -29,7 +29,7 @@ from huggingface_hub import hf_hub_download
|
|
| 29 |
|
| 30 |
def main():
|
| 31 |
|
| 32 |
-
os.system("find / -name nvcc")
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
device = "cuda:0"
|
|
|
|
| 29 |
|
| 30 |
def main():
|
| 31 |
|
| 32 |
+
os.system("find /usr -name nvcc")
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
device = "cuda:0"
|
pre-requirements.txt
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
| 2 |
-
torch==1.13.0
|
| 3 |
-
torchvision==0.14.0
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu113
|
| 2 |
+
torch==1.13.0+cu113
|
| 3 |
+
torchvision==0.14.0+cu113
|