更新时间:2024-04-30 GMT+08:00
分享

transformers调用cuda上的操作,或者执行卡死

问题现象

图1 报错信息

原因分析

transformers库的training_args.py目前适配的是CUDA的部分操作,需要替换为适配NPU的脚本。

处理方法

training_args.py替换为适配NPU的脚本,替换的脚本请见training_args.py

分享:

    相关文档

    相关产品