分享价值
提升价值

最新xformers/CUDA/pytorch版本关系对照表

截止目前最新的Linux&Windows系统平台上pip安装的xformers版本与pytorch版本和CUDA版本关系对照表

xformers pytorch CUDA
0.0.28torch==2.4.1cu118,cu121,cu124
0.0.28.post1torch==2.4.1cu118,cu121,cu124
0.0.27torch==2.3.0cu118,cu121
0.0.27.post2torch==2.3.1cu118,cu121
0.0.27.post1torch==2.3.1cu118,cu121
0.0.26torch==2.2.0cu118,cu121
0.0.25torch==2.2.0cu118,cu121
0.0.24torch==2.2.0cu118,cu121
0.0.23torch==2.1.1
0.0.22torch==2.0.1
0.0.21torch==2.0.1
0.0.20torch==2.0.1
0.0.19torch==2.0.0
0.0.18torch==2.0.0
0.0.17torch==1.13.1

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等版本冲突

附:

xformers whl下载链接:https://download.pytorch.org/whl/xformers/

xformers项目地址:https://github.com/facebookresearch/xformers

赞(0) 打赏
未经允许不得转载:诺瓦软件 » 最新xformers/CUDA/pytorch版本关系对照表

觉得文章对你有帮助就打赏一下作者

非常感谢你的打赏,我们将继续提供更多优质内容,让我们一起创建更加美好的世界!

支付宝扫一扫

微信扫一扫