安装xformers时自动升级PyTorch及依赖？解决方案详解

默认使用pip安装xformers时，会自动升级PyTorch及其依赖——这是因为新版xformers对最新版PyTorch的支持更完善，因此默认安装逻辑会触发PyTorch升级。

要保留当前PyTorch版本，可通过以下两种方式安装对应版本的xformers：

• 指定版本直接安装：xformers的各版本标签对应特定PyTorch版本，比如PyTorch 2.4.1适配xformers==0.0.28.post1，执行以下命令：

  pip install xformers==0.0.28.post1
  

• 源码编译安装：先安装编译依赖ninja，再从对应版本的源码编译安装：

  pip install ninja
  pip install -v -U git+https://github.com/facebookresearch/xformers.git@v0.0.28.post1#egg=xformers
  

内容的提问来源于stack exchange，提问作者Vincent Zhan