AI软件下载
有趣网站推荐及实用软件下载
当前位置：诺瓦小站 问题记录 python 正文

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

2024-09-14 分类：python / 问题记录 评论(0)

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

xformers pytorch CUDA
0.0.32,0.0.32.post1,0.0.32.post2torch==2.8.0cu126,cu128,cu129
0.0.31torch==2.7.1cu118,cu126,cu128
0.0.30torch==2.7.0cu118,cu126,cu128
0.0.29.post2,0.0.29.post3torch==2.6.0cu118,cu124,cu126
0.0.29.post1,0.0.29,0.0.28.post3torch==2.5.1cu118,cu121,cu124
0.0.28.post2torch==2.5.0cu118,cu121,cu124
0.0.28.post1torch==2.4.1cu118,cu121,cu124
0.0.27.post1,0.0.27.post2torch==2.3.1cu118,cu121
0.0.27torch==2.3.0cu118,cu121
0.0.24,0.0.25,0.0.26torch==2.2.0cu118,cu121
0.0.23torch==2.1.1cu118,cu121
0.0.22.post7torch==2.1.0cu118,cu121
0.0.20,0.0.21,0.0.22torch==2.0.1
0.0.18,0.0.19torch==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等版本冲突，最好指定版本安装。例如指定安装最新版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

附：

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

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


AI软件用不了？2元爽玩4090： 立即体验>>

热门大语言模型API免费体验： 立即获取>>

软件催更及1对1人工答疑支持： https://nuowa.net/1806
赞(8) 打赏
软件无法使用？点击查看常见问题说明>>
分享到

相关推荐

最近更新

Claude居然自称“本人”-诺瓦小站

Claude居然自称“本人”

今天在和Claude聊网文写作的时候，Claude回复中居然自称本人，这让我有点震惊。 我经常会和AI聊工具功能对比等话题，ChatGPT、Gemini、Claude这些以前我记得从来没有自称过本人的，好像都是本模型，本工具等等，反正从来没...

赞(1)趣站

免费在线HTML转markdown工具

经常看一些在线文档，想复制给AI参考的话不方便直接复制，转换成markdown再给AI阅读会更简单明了，也方便自己查阅，这里给大家推荐两个在线的免费HTML转markdown网站。 1、https://d1tools.com/tools/c...

赞(0)趣站
答疑咨询/软件定制修改/AI进阶玩法

热门文章

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

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

支付宝扫一扫

微信扫一扫