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
有时候可能会有出入,按照可能的大概版本号上下多试几次即可找到对应版本。
默认是 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