Create unirig_requirements.txt

unirig_requirements.txt

@@ -0,0 +1,47 @@
+# UniRig Dependencies (to be installed into Blender's Python 3.11)
+
+# Numpy (UniRig: 1.26.4)
+numpy==1.26.4
+
+# PyTorch (Matching main env for simplicity, ensure compatibility with UniRig >=2.3.1)
+# Install the version compatible with Python 3.11 and CUDA 12.1
+# Example using index-url, adjust if needed
+torch==2.3.1 --index-url https://download.pytorch.org/whl/cu121
+torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu121
+
+# PyTorch Geometric libraries (for CUDA 12.1, Python 3.11)
+# Note the updated URLs for cp311
+torch-scatter -f https://data.pyg.org/whl/torch-2.3.1+cu121/torch_scatter-latest+pt23cu121-cp311-cp311-linux_x86_64.html
+torch-cluster -f https://data.pyg.org/whl/torch-2.3.1+cu121/torch_cluster-latest+pt23cu121-cp311-cp311-linux_x86_64.html
+
+# Spconv (for CUDA 12.1 - check compatibility with Python 3.11 if issues arise)
+spconv-cu121
+
+# Dependencies from UniRig's official requirements.txt (excluding bpy and flash_attn)
+transformers
+python-box
+einops
+omegaconf
+pytorch_lightning # Or just 'lightning' depending on UniRig's imports
+lightning         # Might be redundant if pytorch_lightning is used
+addict
+timm
+fast-simplification
+# flash_attn removed - requires a cp311 wheel for Python 3.11 + cu121 + torch2.3.1
+trimesh
+open3d
+pyrender
+huggingface_hub
+scipy
+smplx
+chumpy
+opencv-python
+Pillow
+pyquaternion
+hydra-core
+rich
+distinctipy
+scikit-image
+mediapipe
+
+# NOTE: 'bpy' is provided by Blender's environment, do not list it here.