使用torch进行图像分类Resnet50进行训练时报错：RuntimeError:InnerRun:/usr1/03/workspace/j_YJ3Urheg/pytorch/torch_npu/csrc/framework/OpParamMaker.cpp:203 NPU error, error code is 500001

一、问题现象（附报错日志上下文）：
resnet50图像分类任务中，使用transformers库的trainer接口进行模型训练时，在trainer初始化过程中报错：
RuntimeError:InnerRun:/usr1/03/workspace/j_YJ3Urheg/pytorch/torch_npu/csrc/framework/OpParamMaker.cpp:203 NPU error, error code is 500001
当调试定位时发现在trainer初始化过程中将模型向npu迁移中报错：
![输入图片说明](https://foruda.gitee.com/images/1700120714511074968/d5a8c791_11496407.png "屏幕截图")
参数device：
![输入图片说明](https://foruda.gitee.com/images/1700120777604621282/a6003886_11496407.png "屏幕截图")
出错位置：
![输入图片说明](https://foruda.gitee.com/images/1700125224780607722/c82345d3_11496407.png "屏幕截图")
参数module.weight.data
![输入图片说明](https://foruda.gitee.com/images/1700125248607108174/0c302e2b_11496407.png "屏幕截图")

于是测试结果如下：
![输入图片说明](https://foruda.gitee.com/images/1700121297791808914/b06c1606_11496407.png "屏幕截图")

二、软件版本:
-- CANN 版本 (e.g., CANN 3.0.x，5.x.x):  7.0.RC1
--Pytorch版本: 2.1.0
--Python 版本 (e.g., Python 3.7.5):Python 3.8.0
-- MindStudio版本 (e.g., MindStudio 2.0.0 (beta3)):无
--操作系统版本 (e.g., Ubuntu 18.04):CentOS Linux release 7.9.2009 (AltArch)
--驱动和固件：Ascend-hdk-910-npu-driver_23.0.rc3_linux-aarch64.run     Ascend-hdk-910-npu-firmware_7.0.0.5.242.run

Ascend/pytorch
暂停

内容风险标识

评论 (5)

Ascend/pytorch暂停 .gitee-modal { width: 500px !important; }

内容风险标识