torch2.5

source /activate.sh && pip install torch-scatter -f https://data.pyg.org/whl/torch-2.5.0+cu124.html

torch 高版本：

# 确保在正确的环境中
conda activate hunyuan3d21  # 或你的目标环境

# 安装编译依赖
pip install pybind11 ninja cmake

# 设置环境变量
export TORCH_CUDA_ARCH_LIST="6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0;12.0"
export CUDA_NVCC_FLAGS="-allow-unsupported-compiler"

# 克隆并编译
git clone https://github.com/rusty1s/pytorch_scatter.git
cd pytorch_scatter

# 清理旧构建
rm -rf build dist *.egg-info

# 编译安装
python setup.py clean
python setup.py build
python setup.py install

# 或使用 pip
pip install .