最近復現的一些倉庫用 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/xformers.git@v0.28.0#egg=xformers
@ 後面的填 tag 。