Yolov5_for_PyTorch_v6.0推理和输出报错

一、问题现象：
detect  显示cuda不可用，指令也无法调用输出；export没有onnx输出，加上模块包还是报错，无法输出结果。
![输入图片说明](https://foruda.gitee.com/images/1698634630937674124/cf0876d9_13622919.png "屏幕截图")
![输入图片说明](https://foruda.gitee.com/images/1698634661172015867/11b83613_13622919.png "屏幕截图")
![输入图片说明](https://foruda.gitee.com/images/1698634675206985952/064178c8_13622919.png "屏幕截图")
raise RuntimeError('Attempting to deserialize object on a CUDA '
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
THPModule_npu_shutdown success.

ONNX: export failure: Node (Resize_117)'s input 1 is marked single but has an empty string in the graph
==> Context: Bad node spec: input: "238" input: "" input: "525" output: "243" name: "Resize_117" op_type: "Resize" attribute { name: "coordinate_transformation_mode" s: "asymmetric" type: STRING } attribute { name: "cubic_coeff_a" f: -0.75 type: FLOAT } attribute { name: "mode" s: "nearest" type: STRING } attribute { name: "nearest_mode" s: "floor" type: STRING }

二、软件版本:
pytorch1 8 1:pytorch 1.8.1-cann 6.0.1python 3.8-euler 2.8-v2

Ascend/modelzoo-GPL
暂停

内容风险标识

评论 (0)

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

内容风险标识