华为云AI开发平台ModelArtstransformers调用cuda 上的操作,或者执行卡死_云淘科技

问题现象

图1 报错信息

原因分析

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

处理方法

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

父主题: 常见问题

同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)

内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家