錯(cuò)誤現(xiàn)象
在使用 sd-script訓(xùn)練自己的Lora模型時(shí),可能會(huì)出現(xiàn)“Error no kernel image is available for execution on the device at line 167 ”的錯(cuò)誤:
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
錯(cuò)誤原因
這是因?yàn)榕f的顯卡 例如10系列不支持unit8類型文章來源:http://www.zghlxwxcb.cn/news/detail-505897.html
解決方法
,需要在 訓(xùn)練的配置文件中,修改:文章來源地址http://www.zghlxwxcb.cn/news/detail-505897.html
# use 8bit adam optimizer | 使用 8bit adam 優(yōu)化器節(jié)省顯存,默認(rèn)啟用。
# 部分 10 系老顯卡無法使用,修改為 0 禁用。
$use_8bit_adam = 0
$use_lion = 0 # use lion optimizer | 使用 Lion 優(yōu)化器
到了這里,關(guān)于【Lora模型訓(xùn)練過程報(bào)錯(cuò)】Error no kernel image is available for execution on the device at line的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!