文档首页/
AI开发平台ModelArts/
最佳实践/
GPU业务迁移至昇腾训练推理/
基于LLM模型的GPU训练业务迁移至昇腾指导/
常见问题/
transformers调用cuda上的操作，或者执行卡死
更新时间：2024-12-30 GMT+08:00
transformers调用cuda上的操作，或者执行卡死
问题现象
图1 报错信息
原因分析
transformers库的training_args.py目前适配的是CUDA的部分操作，需要替换为适配NPU的脚本。
处理方法
training_args.py替换为适配NPU的脚本，替换的脚本请见training_args.py。
父主题： 常见问题
