stable diffusion model训练遇到的问题【No module named ‘triton‘】

一天早晨过来,发现昨天还能跑的diffusion代码,突然出现了【No module named 'triton'】的问题,导致本就不富裕的显存和优化速度雪上加霜,因此好好探究了解决方案。

首先是原因,由于早晨过来发现【电脑重启】导致了【训练终止】(美好的心情从看到windows更新结束),基本可以判定是由于windows更新,以及所编译的triton的windows版本的原因,最终解决方案如下:

1、第一步,按顺序卸载triton、xformers、torch

bash 复制代码
pip uninstall triton
pip uninstall xformers
pip uninstall torch

2、第二步,利用huggingface上直接下载到本地的triton-2.0.0-cp310-cp310-win_amd64.whl进行pip安装,安装triton。具体的windows下编译好的triton的网址为 https://huggingface.co/datasets/ArrayCats/triton-2.0.0-cp310-cp310-win_amd64/resolve/main/triton-2.0.0-cp310-cp310-win_amd64.whl

bash 复制代码
pip install triton-2.0.0-cp310-cp310-win_amd64.whl

3、第三步,安装xformers

bash 复制代码
pip install xformers

4、第四步,在安装triton时,下载的torch为cpu版本,无法运用cuda和xformers,因此根据卸载前的版本或者xformers的报错,安装对应版本的torch,我个人所需要的是torch==2.0.1+cu118,所以在pytorch官网的安装命令上进行了修改,指定了torch版本

bash 复制代码
pip3 install torch==2.0.1+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
相关推荐
LittleNyima8 小时前
【代码解读】阿里最新开源视频生成模型 Wan 2.1 实现解析
人工智能·stable diffusion·aigc·音视频
AIGC安琪13 小时前
AI绘画(Stable Diffusion)喂饭级教程
人工智能·ai作画·stable diffusion·aigc·美女
乱世刀疤14 小时前
AI绘画软件Stable Diffusion详解教程(2):Windows系统本地化部署操作方法(专业版)
ai作画·stable diffusion
Donvink21 小时前
【AIGC系列】3:Stable Diffusion模型介绍
人工智能·深度学习·语言模型·stable diffusion·aigc·transformer
学不会lostfound2 天前
五、AIGC大模型_04LLaMA-Factory基础知识与SFT实战
lora·深度学习模型·自我认知·llama-factory·sft微调·aigc实战
PeterClerk3 天前
AIGC-Stable Diffusion模型介绍
图像处理·人工智能·python·自然语言处理·stable diffusion·aigc
农夫山泉2号5 天前
【个人开源】——从零开始在高通手机上部署sd(二)
stable diffusion·智能手机·sd·高通·qnn
后知后觉5 天前
windows 安装 stable diffusion
stable diffusion
农夫山泉2号5 天前
【个人开源】——从零开始在高通手机上部署sd(一)
stable diffusion·智能手机·sd·高通·qnn
AIGC安琪6 天前
【Stable Diffusion】SD迎来动画革命,AnimateDiff快速出图
人工智能·网络协议·tcp/ip·stable diffusion·aigc