【工具】最新xformers/CUDA/pytorch版本关系对照表

cainiaoshileyuan

于 2025-05-16 10:16:59 发布

阅读量346

点赞数 4

文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/cainiaoshileyuan/article/details/148000602

版权

截止目前最新的Linux&Windows系统平台上pip安装的xformers版本与pytorch版本和CUDA版本关系对照表
在这里插入图片描述

xformers v0.0.28版本Linux&Windows平台安装pip命令参考：

# [linux only] cuda 11.8 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
# [linux only] cuda 12.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121
# [linux & win] cuda 12.4 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu124
# [linux only] (EXPERIMENTAL) rocm 6.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/rocm6.1

安装xformers的时候可以和torch一起安装，不然的话torch还会再重新安装一次，而且torch会自动选择到当前xformers支持到的最新版本，可能会与你当前想要的torch版本不一样导致torchvision等版本冲突，最好指定版本安装。例如指定安装最新版torch 2.5.1和xformers 0.0.28.post3和cu124,可以使用如下安装命令：

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 xformers==0.0.28.post3 --index-url https://download.pytorch.org/whl/cu124
# CUDA 11.8
pip install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 xformers==0.0.23 --index-url https://download.pytorch.org/whl/cu118