xformers 安装相关问题_xformers版本

xformers版本与pytorch版本对应

xformer直接pip install 需要最新版本的pytorch，从 pypi release history 可推算出各个pytorch版本对应的xformers版本。

torch==2.2.2 xformers==0.0.25.post1
torch==2.2.1 xformers==0.0.25
1
2

有时候可能会有出入，按照可能的大概版本号上下多试几次即可找到对应版本。

默认是 cuda12.1 ， 想要安装cuda11.8 版本，要

# cuda 11.8 version
pip3 install -U xformers==0.0.25.post1 --index-url https://download.pytorch.org/whl/cu118
# cuda 12.1 version
pip3 install -U xformers==0.0.25.post1 --index-url https://download.pytorch.org/whl/cu121
1
2
3
4