Update requirements.txt

requirements.txt

@@ -1,3 +1,13 @@
+# --- 环境配置 ---
+--extra-index-url https://download.pytorch.org/whl/cu124
+--find-links https://data.pyg.org/whl/torch-2.5.0+cu124.html
+
+# --- 核心依赖 (ZeroGPU 已预装 Torch, 但声明版本可确保兼容性) ---
+torch==2.5.1+cu124
+torchvision
+torch_cluster
+
+# --- 其他 3D/图像处理库 ---
 scikit-learn
 gpustat
 nvitop
@@ -17,6 +27,4 @@ imageio-ffmpeg
 pyrender
 deepspeed
 wandb[media]
-colormaps
-torchvision
-torch_cluster @ https://github.com/OutofAi/PyTorch3D-wheels/releases/download/0.7.8/torch_cluster-1.6.3.postHopper271126-cp310-cp310-linux_x86_64.whl
+colormaps