Spaces:
Running
on
Zero
Running
on
Zero
Stanislaw Szymanowicz
commited on
Commit
•
693a19d
1
Parent(s):
4237c2d
Install cuda-compatible torch
Browse files- pre-requirements.txt +1 -0
pre-requirements.txt
CHANGED
@@ -1,2 +1,3 @@
|
|
|
|
1 |
torch==1.13.0
|
2 |
torchvision==0.14.0
|
|
|
1 |
+
--extra-index-url https://download.pytorch.org/whl/cu113
|
2 |
torch==1.13.0
|
3 |
torchvision==0.14.0
|