更新时间:2024-04-30 GMT+08:00
transformers调用cuda上的操作,或者执行卡死
问题现象
图1 报错信息
原因分析
transformers库的training_args.py目前适配的是CUDA的部分操作,需要替换为适配NPU的脚本。
处理方法
training_args.py替换为适配NPU的脚本,替换的脚本请见training_args.py。
父主题: 常见问题