[代码调试]安装Text2Image(stablediffusion)模型环境的踩坑记录

文章目录 一、xFormers版本问题1、先确认下自己torch版本所对应的cuda版本2.安装对应版本(1) 到对应官网下载包本地安装(2)代码安装 二、attn_mask尺寸错误三、diffusers四、Huggingface 配置基于stable diffusion预训练模型的环境，记录踩坑记录

一、xFormers版本问题

xformers是用来加速推理的，但是和pytorch的版本有兼容问题，需要安装对应的版本，否则会报错。

1、先确认下自己torch版本所对应的cuda版本

可以使用以下代码进行查看

import torch print(torch.version.cuda) 2.安装对应版本

安装方式有三种

(1) 到对应官网下载包本地安装

如果是pip安装到这里

download.pytorch.org/whl/cu118

如果是conda安装到这里

anaconda.org/xformers/xformers/files

上面有不同版本及对应关系，如果不知道当前版本需要安装哪个xFormers版本，可以在这找找

(2)代码安装

上面网站conda 查找到需要安装的版本，直接pip安装

pip3 install xformers==0.0.26.post1

或者直接代码安装

pip3 install -U xformers --index-url download.pytorch.org/whl/cu118

后面的cu118换成自己的cuda版本，但是有可能torch版本会给你换掉，注意看下

二、attn_mask尺寸错误

出现这个问题RuntimeError: The shape of the 2D attn_mask is torch.Size([77, 77]), but should be (1, 1)

一般需要查看transformers 和 open-clip-torch 版本，让其和官方requirements.txt一致。反正就是升降级

三、diffusers

如果发现找不到diffusers，可以尝试安装一下更低or更高的版本，最好和官方requirements.txt一致

四、Huggingface

如果发现找不到Huggingface，可以尝试安装一下更低or更高的版本，最好和官方requirements.txt一致