一天早晨过来，发现昨天还能跑的diffusion代码，突然出现了【No module named ‘triton’】的问题，导致本就不富裕的显存和优化速度雪上加霜，因此好好探究了解决方案。
首先是原因，由于早晨过来发现【电脑重启】导致了【训练终止】（美好的心情从看到windows更新结束），基本可以判定是由于windows更新，以及所编译的triton的windows版本的原因，最终解决方案如下：
1、第一步，按顺序卸载triton、xformers、torch
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
pip install triton-2.0.0-cp310-cp310-win_amd64.whl
3、第三步，安装xformers文章来源：https://www.toymoban.com/news/detail-738929.html
pip install xformers
4、第四步，在安装triton时，下载的torch为cpu版本，无法运用cuda和xformers，因此根据卸载前的版本或者xformers的报错，安装对应版本的torch，我个人所需要的是torch==2.0.1+cu118，所以在pytorch官网的安装命令上进行了修改，指定了torch版本文章来源地址https://www.toymoban.com/news/detail-738929.html
pip3 install torch==2.0.1+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
到了这里，关于stable diffusion model训练遇到的问题【No module named ‘triton‘】的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
