华为云AI开发平台ModelArtstransformers调用cuda 上的操作,或者执行卡死_云淘科技
问题现象
图1 报错信息
原因分析
transformers库的tranining_args.py目前适配的是CUDA的部分操作,需要替换为适配NPU的脚本。
处理方法
tranining_args.py替换为适配NPU的脚本,替换的脚本请见training_args.py。
父主题: 常见问题
同意关联代理商云淘科技,购买华为云产品更优惠(QQ 78315851)
内容没看懂? 不太想学习?想快速解决? 有偿解决: 联系专家