墨斋记

Good Luck To You!

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

xformerspytorchCUDA
v0.0.30torch==2.7.0cu118,cu126,cu128
v0.0.29.post3torch==2.6.0,12.6,
v0.0.29.post2torch==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.26torch==2.2.0cu118,cu121
0.0.25torch==2.2.0cu118,cu121
0.0.24torch==2.2.0cu118,cu121
0.0.23torch==2.1.1cu118,cu121
0.0.22.post7torch==2.1.0cu118,cu121
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  
# [linux only] cuda 12.1 version
pip3 install -U xformers --index-url  
# [linux & win] cuda 12.4 version
pip3 install -U xformers --index-url  
# [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


发表评论:

Powered By Z-BlogPHP 1.7.3

冀公网安备13019902000834
冀ICP备18019600号-1
Copyright CC Some Rights Reserved.Contact Email:chengyingbo88@163.com