cann 7.0.1  torch_npu 1.11

File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1110, in _call_impl 

    return forward_call(*input, **kwargs) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/detectron2/modeling/proposal_generator/rpn.py", line 443, in forward 

    proposals = self.predict_proposals( 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context 

    return func(*args, **kwargs) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/detectron2/modeling/proposal_generator/rpn.py", line 469, in predict_proposals 

    return find_top_rpn_proposals( 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/detectron2/modeling/proposal_generator/proposal_utils.py", line 107, in find_top_rpn_proposals 

    keep = batched_nms(boxes.tensor, scores_per_img, lvl, nms_thresh) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/detectron2/layers/nms.py", line 19, in batched_nms 

    return box_ops.batched_nms(boxes, scores, idxs, iou_threshold) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torchvision/ops/boxes.py", line 72, in batched_nms 

    return _batched_nms_vanilla(boxes, scores, idxs, iou_threshold) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torch/jit/_trace.py", line 1118, in wrapper 

    return fn(*args, **kwargs) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torchvision/ops/boxes.py", line 108, in _batched_nms_vanilla 

    curr_keep_indices = nms(boxes[curr_indices], scores[curr_indices], iou_threshold) 

  File "/root/anaconda3/envs/cmcc/lib/python3.9/site-packages/torchvision/ops/boxes.py", line 40, in nms 

    return torch.ops.torchvision.nms(boxes, scores, iou_threshold) 

NotImplementedError: 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, Named, Conjugate, Negative, ZeroTensor, ADInplaceOrView, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, AutogradLazy, AutogradXPU, AutogradMLC, AutogradHPU, Tracer, AutocastCPU, Autocast, Batched, VmapMode, Functionalize].