【踩坑】HuggingFace在Spaces中安装torch-scatter

如果本文帮助到了你，欢迎[点赞、收藏、关注]哦~

问题描述

        Hugging Face Spaces 在 build 阶段安装 requirements.txt 时是顺序安装的，因此当它安装到 torch-scatter 时，PyTorch 可能还没装好（或者环境里版本不匹配），所以就报错「找不到 torch」。

解决方法

        用 pip install 指定 PyTorch 版本和 torch-scatter 的兼容 wheel。torch-scatter 在 PyPI 上的很多 wheel 不完整，官方推荐用源码编译或者从 PyG 的 whl 仓库 安装。

        因此，在 requirements.txt 开头写成这样：

--find-links https://data.pyg.org/whl/torch-2.0.1+cpu.html
torch==2.0.1
torch-scatter==2.1.2
torchvision==0.15.2

        如果是GPU的话，就写成：

--find-links https://data.pyg.org/whl/torch-2.0.1+cu117.html
torch-scatter==2.1.2

        这样可以确保 torch 先装，再装 torch-scatter，并且 wheel 来源正确。