xFormers pip 安装

最近复现的一些仓库用 xFormers 的很多，在 arm 的 aarch64 下安装所有和 CUDA 相关的库都不是非常方便，这里记录一下。

参考：
https://github.com/facebookresearch/xformers
https://blog.csdn.net/x1131230123/article/details/139231686

首先要确定版本，xFormers 高度绑定 pytorch 和 CUDA 版本，基本上一一对应。

如果仓库的 requirements.txt 给了，那就最好，只要保证其它库版本一致。

如果 pytorch 有改动，那么要求找一下 xFormers 对应的版本。

比如打开这个：
https://github.com/facebookresearch/xformers/tree/v0.0.28

选择tag，在 README.md 可以看到

二分查找找一下 pytorch版本需要对应什么，比如 pytorch2.1.0 对应 0.0.23(这个是对应 pytorch2.1.1 的，找不到 2.1.0 的)

如果 x86，直接 pip 安装即可：

pip3 install -U xformers==0.28.0 --index-url https://download.pytorch.org/whl/cu118

如果 arm，从源码安装：

pip install -v -U git+https://github.com/facebookresearch/[email protected]#egg=xformers

@ 后面的填 tag 。