Update requirements.txt

requirements.txt

@@ -1,4 +1,8 @@
-# Other Python dependencies
+# 1. 降级 setuptools 解决编译环境问题
+setuptools<=69.5.1
+
+# 2. 基础依赖库
+numpy==1.23.5
 scipy==1.13.1
 scikit-image==0.21.0
 scikit-learn==1.2.2
@@ -9,35 +13,43 @@ PyYAML==6.0.2
 tqdm==4.67.1
 absl-py==2.1.0
 tensorboard==2.17.0
-PyOpenGL==3.1.0
-pyrender==0.1.45
-trimesh==3.22.0
 click==8.1.7
 omegaconf==2.2.3
-segmentation_models_pytorch
-timm==1.0.15
 psutil==5.9.5
 lmdb==1.4.1
-einops==0.8.1
-kornia==0.6.7
 gdown==5.2.0
 plyfile==1.0.3
 natsort
-mmcv==1.7.0
-xformers==0.0.28
 matplotlib==3.9.1.post1
+imageio_ffmpeg
+
+# 3. 三维与多模态通用工具（不强依赖 C++ 框架层的）
+PyOpenGL==3.1.0
+pyrender==0.1.45
+trimesh==3.22.0
+segmentation_models_pytorch
+timm==1.0.15
+einops==0.8.1
+kornia==0.6.7
 mediapipe==0.10.5
-numpy==1.23.5
 open-clip-torch
-imageio_ffmpeg
-spaces
 transformers==4.41.2
 diffusers==0.29.2
-# Install torch & torchvision from specific CUDA-enabled wheels (CUDA 12.1, Python 3.10)
+spaces
+
+# =========================================================
+# 4. 后置安装区域：核心框架、编译工具及强依赖框架的扩展库
+# =========================================================
+ninja
+
+# PyTorch 核心与全家桶
 https://download.pytorch.org/whl/cu121/torch-2.4.1%2Bcu121-cp310-cp310-linux_x86_64.whl
 https://download.pytorch.org/whl/cu121/torchvision-0.19.1%2Bcu121-cp310-cp310-linux_x86_64.whl
 
-# PyTorch3D from custom index
+# PyTorch3D 官方预编译包
 -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt241/download.html
 pytorch3d
-ninja
+
+# 依赖 PyTorch 2.4.1 且需要动态编译的扩展库
+xformers==0.0.28
+mmcv==1.7.0