【Lora模型训练过程报错】Error no kernel image is available for execution on the device at line

【Lora模型训练过程报错】Error no kernel image is available for execution on the device at line

    正在检查是否收录...

错误现象

在使用 sd-script训练自己的Lora模型时,可能会出现“Error no kernel image is available for execution on the device at line 167 ”的错误:

steps: 0%| | 0/5040 [00:00<?, ?it/s]epoch 1/20 Error no kernel image is available for execution on the device at line 167 in file D:\ai\tool\bitsandbytes\csrc\ops.cu Traceback (most recent call last): File "S:\miniconda\envs\stable\lib\runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "S:\miniconda\envs\stable\lib\runpy.py", line 86, in _run_code exec(code, run_globals) File "S:\01code\stable_diff\lora-scripts\venv\Scripts\accelerate.exe\__main__.py", line 7, in <module> File "S:\01code\stable_diff\lora-scripts\venv\lib\site-packages\accelerate\commands\accelerate_cli.py", line 45, in main 

错误原因

这是因为旧的显卡 例如10系列不支持unit8类型

解决方法

,需要在 训练的配置文件中,修改:

 # use 8bit adam optimizer | 使用 8bit adam 优化器节省显存,默认启用。 # 部分 10 系老显卡无法使用,修改为 0 禁用。 $use_8bit_adam = 0 $use_lion = 0 # use lion optimizer | 使用 Lion 优化器 

codescripticonunitcliurl
  • 本文作者:李琛
  • 本文链接: https://wapzz.net/post-4406.html
  • 版权声明:本博客所有文章除特别声明外,均默认采用 CC BY-NC-SA 4.0 许可协议。
本站部分内容来源于网络转载,仅供学习交流使用。如涉及版权问题,请及时联系我们,我们将第一时间处理。
文章很赞!支持一下吧 还没有人为TA充电
为TA充电
还没有人为TA充电
0
  • 支付宝打赏
    支付宝扫一扫
  • 微信打赏
    微信扫一扫
感谢支持
文章很赞!支持一下吧
关于作者
2.3W+
5
0
1
WAP站长官方

2024年最热门的家居科技趋势

上一篇

GaussDB(DWS)中的分布式死锁问题实践

下一篇
  • 复制图片
按住ctrl可打开默认菜单