Spaces:
Running
on
Zero
Running
on
Zero
Ruining Li
commited on
Commit
·
5891b5e
1
Parent(s):
188bd1d
Resolve dependency
Browse files- requirements.txt +8 -9
requirements.txt
CHANGED
|
@@ -1,14 +1,17 @@
|
|
| 1 |
-
#
|
| 2 |
-
--index-url https://download.pytorch.org/whl/cu124
|
|
|
|
|
|
|
|
|
|
| 3 |
|
| 4 |
torch==2.4.0
|
| 5 |
torchvision==0.19.0
|
| 6 |
torchaudio==2.4.0
|
| 7 |
-
|
| 8 |
-
# Extra wheel index for torch-scatter (depends on torch 2.4.0+cu124)
|
| 9 |
-
--find-links https://data.pyg.org/whl/torch-2.4.0+cu124.html
|
| 10 |
torch-scatter
|
| 11 |
|
|
|
|
|
|
|
|
|
|
| 12 |
yacs
|
| 13 |
trimesh
|
| 14 |
scikit-image
|
|
@@ -26,7 +29,3 @@ simple_parsing
|
|
| 26 |
arrgh
|
| 27 |
open3d
|
| 28 |
vtk
|
| 29 |
-
|
| 30 |
-
omegaconf
|
| 31 |
-
diffusers
|
| 32 |
-
scipy
|
|
|
|
| 1 |
+
# Keep PyPI as default, add the PyTorch CUDA wheels as an extra index:
|
| 2 |
+
--extra-index-url https://download.pytorch.org/whl/cu124
|
| 3 |
+
|
| 4 |
+
# Extra wheel index for torch-scatter
|
| 5 |
+
--find-links https://data.pyg.org/whl/torch-2.4.0+cu124.html
|
| 6 |
|
| 7 |
torch==2.4.0
|
| 8 |
torchvision==0.19.0
|
| 9 |
torchaudio==2.4.0
|
|
|
|
|
|
|
|
|
|
| 10 |
torch-scatter
|
| 11 |
|
| 12 |
+
psutil
|
| 13 |
+
lightning==2.2
|
| 14 |
+
h5py
|
| 15 |
yacs
|
| 16 |
trimesh
|
| 17 |
scikit-image
|
|
|
|
| 29 |
arrgh
|
| 30 |
open3d
|
| 31 |
vtk
|
|
|
|
|
|
|
|
|
|
|
|