科技网站内容设计,郑州网站设计推荐,余姚公司网站建设,3g 手机网站一天早晨过来#xff0c;发现昨天还能跑的diffusion代码#xff0c;突然出现了【No module named ‘triton’】的问题#xff0c;导致本就不富裕的显存和优化速度雪上加霜#xff0c;因此好好探究了解决方案。
首先是原因#xff0c;由于早晨过来发现【电脑重启】导致了【…一天早晨过来发现昨天还能跑的diffusion代码突然出现了【No module named ‘triton’】的问题导致本就不富裕的显存和优化速度雪上加霜因此好好探究了解决方案。
首先是原因由于早晨过来发现【电脑重启】导致了【训练终止】美好的心情从看到windows更新结束基本可以判定是由于windows更新以及所编译的triton的windows版本的原因最终解决方案如下
1、第一步按顺序卸载triton、xformers、torch
pip uninstall triton
pip uninstall xformers
pip uninstall torch2、第二步利用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.whl3、第三步安装xformers
pip install xformers4、第四步在安装triton时下载的torch为cpu版本无法运用cuda和xformers因此根据卸载前的版本或者xformers的报错安装对应版本的torch我个人所需要的是torch2.0.1cu118所以在pytorch官网的安装命令上进行了修改指定了torch版本
pip3 install torch2.0.1cu188 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
