tensorflow2 使用npu训练,显核无占用,显存占用低,cpu占用高
收藏回复举报
tensorflow2 使用npu训练,显核无占用,显存占用低,cpu占用高
t('forum.solved') 已解决
新人帖
发表于2024-12-17 16:09:55
0 查看

使用tf2,具体代码无法给出,但是顶部代码

import npu_device as npu
npu.open().as_default()

运行正常,无报错。

整个训练流程没有报错,可见cpu占用接近100%,gpu显核无占用,显存占用仅6%,训练速度很慢,日志中存在

Warning: set_tiling_params does not take effect because get_op_mode is dynamic

应当如何排查

本帖最后由 匿名用户2024/12/17 17:20:42 编辑

我要发帖子