多模态模型运行时提示Warning: CAUTION: The operator 'torchvision::xxxx' is not currently supported on the NPU backend and
收藏回复举报
多模态模型运行时提示Warning: CAUTION: The operator 'torchvision::xxxx' is not currently supported on the NPU backend and
t('forum.solved') 已解决
发表于2025-04-16 22:40:06
0 查看

【问题描述】

模型运行时提示:deform_conv2d和nms算子NPU上不支持,运行在了CPU上,但程序本身并没有报错

Warning: CAUTION: The operator 'torchvision::deform_conv2d' is not currently supported on the NPU backend and will fall back to run on the CPU.

Warning: CAUTION: The operator 'torchvision::nms’ is not currently supported on the NPU backend and will fall back to run on the cpu

【版本信息】

CANN:8.0.RC1

Torch:2.2.0

Torch_npu:2.2.0

Torchvision:0.17.0

Torchvision_npu: xxx

我要发帖子