Spaces:
Runtime error
Runtime error
update
#13
by
zhuoyanglin
- opened
- requirements.txt +10 -21
requirements.txt
CHANGED
|
@@ -1,23 +1,8 @@
|
|
| 1 |
-
# torch
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
# pyrender
|
| 5 |
-
# opencv-python
|
| 6 |
-
# yacs
|
| 7 |
-
# scikit-image
|
| 8 |
-
# einops
|
| 9 |
-
# timm
|
| 10 |
-
# OmegaConf
|
| 11 |
-
|
| 12 |
-
--extra-index-url https://download.pytorch.org/whl/cu116
|
| 13 |
-
torch==1.13.1+cu116
|
| 14 |
-
torchvision==0.14.1+cu116
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
# --extra-index-url https://download.pytorch.org/whl/cpu
|
| 18 |
-
# torch==1.13.1+cpu
|
| 19 |
-
# torchvision==0.14.1+cpu
|
| 20 |
|
|
|
|
| 21 |
pytorch-lightning
|
| 22 |
smplx==0.1.28
|
| 23 |
opencv-python
|
|
@@ -25,11 +10,15 @@ yacs
|
|
| 25 |
scikit-image
|
| 26 |
einops
|
| 27 |
timm
|
| 28 |
-
|
| 29 |
trimesh
|
| 30 |
pyopengl==3.1.0
|
| 31 |
pyglet
|
| 32 |
PyOpenGL
|
| 33 |
PyOpenGL_accelerate
|
| 34 |
numpy==1.23.3
|
| 35 |
-
shapely
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# --- PyTorch 2.x 组合，解决 detectron2 使用 @torch.compiler.disable 的不兼容 ---
|
| 2 |
+
torch>=2.1
|
| 3 |
+
torchvision>=0.16
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
+
# ---- 核心依赖（保持与原仓库一致/等价）----
|
| 6 |
pytorch-lightning
|
| 7 |
smplx==0.1.28
|
| 8 |
opencv-python
|
|
|
|
| 10 |
scikit-image
|
| 11 |
einops
|
| 12 |
timm
|
| 13 |
+
omegaconf
|
| 14 |
trimesh
|
| 15 |
pyopengl==3.1.0
|
| 16 |
pyglet
|
| 17 |
PyOpenGL
|
| 18 |
PyOpenGL_accelerate
|
| 19 |
numpy==1.23.3
|
| 20 |
+
shapely
|
| 21 |
+
|
| 22 |
+
# ---- Detectron2（预编译轮子，匹配 Torch 2.1 + CUDA 11.8）----
|
| 23 |
+
--find-links https://dl.fbaipublicfiles.com/detectron2/wheels/cu118/torch2.1/index.html
|
| 24 |
+
detectron2
|