求助!Could not run 'torchvision::nms' with arguments from the 'XLA' backend.
收藏回复举报
求助!Could not run 'torchvision::nms' with arguments from the 'XLA' backend.
t('forum.solved') 已解决
发表于2025-04-10 09:18:43
0 查看

昇腾 300Tpro

CANN 7.0

pytorch 2.0.1

ubuntu docker环境可正常运行其他代码,相关环境配置无错误。

docker运行ssd/frcnn/yolox相关代码时报如下错误

Could not run 'torchvision::nms' with arguments from the 'XLA' backend. This could be because the operator doesn't exist for this backend, or was omitted during the selective/custom build process (if using custom build). If you are a Facebook employee using PyTorch on mobile, please visit https://fburl.com/ptmfixes for possible resolutions. 'torchvision::nms' is only available for these backends: [CPU, QuantizedCPU, BackendSelect, Python, FuncTorchDynamicLayerBackMode, Functionalize, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradMPS, AutogradXPU, AutogradHPU, AutogradLazy, AutogradMeta, Tracer, AutocastCPU, AutocastCUDA, FuncTorchBatched, FuncTorchVmapMode, Batched, VmapMode, FuncTorchGradWrapper, PythonTLSSnapshot, FuncTorchDynamicLayerFrontMode, PythonDispatcher].

CPU: registered at /root/vision/torchvision/csrc/ops/cpu/nms_kernel.cpp:112 [kernel]

QuantizedCPU: registered at /root/vision/torchvision/csrc/ops/quantized/cpu/qnms_kernel.cpp:124 [kernel]

BackendSelect: fallthrough registered at /root/pytorch/aten/src/ATen/core/BackendSelectFallbackKernel.cpp:3 [backend fallback]

Python: registered at /root/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:144 [backend fallback]

FuncTorchDynamicLayerBackMode: registered at /root/pytorch/aten/src/ATen/functorch/DynamicLayer.cpp:491 [backend fallback]

Functionalize: registered at /root/pytorch/aten/src/ATen/FunctionalizeFallbackKernel.cpp:280 [backend fallback]

Named: registered at /root/pytorch/aten/src/ATen/core/NamedRegistrations.cpp:7 [backend fallback]

Conjugate: registered at /root/pytorch/aten/src/ATen/ConjugateFallback.cpp:17 [backend fallback]

Negative: registered at /root/pytorch/aten/src/ATen/native/NegateFallback.cpp:19 [backend fallback]

ZeroTensor: registered at /root/pytorch/aten/src/ATen/ZeroTensorFallback.cpp:86 [backend fallback]

ADInplaceOrView: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:63 [backend fallback]

AutogradOther: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:30 [backend fallback]

AutogradCPU: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:34 [backend fallback]

AutogradCUDA: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:42 [backend fallback]

AutogradXLA: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:46 [backend fallback]

AutogradMPS: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:54 [backend fallback]

AutogradXPU: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:38 [backend fallback]

AutogradHPU: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:67 [backend fallback]

AutogradLazy: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:50 [backend fallback]

AutogradMeta: fallthrough registered at /root/pytorch/aten/src/ATen/core/VariableFallbackKernel.cpp:58 [backend fallback]

Tracer: registered at /root/pytorch/torch/csrc/autograd/TraceTypeManual.cpp:294 [backend fallback]

AutocastCPU: fallthrough registered at /root/pytorch/aten/src/ATen/autocast_mode.cpp:487 [backend fallback]

AutocastCUDA: fallthrough registered at /root/pytorch/aten/src/ATen/autocast_mode.cpp:354 [backend fallback]

FuncTorchBatched: registered at /root/pytorch/aten/src/ATen/functorch/LegacyBatchingRegistrations.cpp:815 [backend fallback]

FuncTorchVmapMode: fallthrough registered at /root/pytorch/aten/src/ATen/functorch/VmapModeRegistrations.cpp:28 [backend fallback]

Batched: registered at /root/pytorch/aten/src/ATen/LegacyBatchingRegistrations.cpp:1073 [backend fallback]

VmapMode: fallthrough registered at /root/pytorch/aten/src/ATen/VmapModeRegistrations.cpp:33 [backend fallback]

FuncTorchGradWrapper: registered at /root/pytorch/aten/src/ATen/functorch/TensorWrapper.cpp:210 [backend fallback]

PythonTLSSnapshot: registered at /root/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:152 [backend fallback]

FuncTorchDynamicLayerFrontMode: registered at /root/pytorch/aten/src/ATen/functorch/DynamicLayer.cpp:487 [backend fallback]

PythonDispatcher: registered at /root/pytorch/aten/src/ATen/core/PythonFallbackKernel.cpp:148 [backend fallback]

我要发帖子