当前位置:   article > 正文

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

error no kernel image is available for execution on the device at line 167 i

错误现象

在使用 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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9

错误原因

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

解决方法

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

 # use 8bit adam optimizer | 使用 8bit adam 优化器节省显存,默认启用。
 # 部分 10 系老显卡无法使用,修改为 0 禁用。
$use_8bit_adam = 0
$use_lion = 0 # use lion optimizer | 使用 Lion 优化器
  • 1
  • 2
  • 3
  • 4
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/101179
推荐阅读
相关标签
  

闽ICP备14008679号