RuntimeError: No such operator torchvision::nms

遇到问题：

Traceback (most recent call last):
  File "efficientdet_test.py", line 79, in <module>
    threshold, iou_threshold)
  File "/home/ubuntu/bigdisk/part1/Yet-Another-EfficientDet-Pytorch/utils/utils.py", line 109, in postprocess
    anchors_nms_idx = batched_nms(transformed_anchors_per, scores_per[:, 0], classes_, iou_threshold=iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 76, in batched_nms
    keep = nms(boxes_for_nms, scores, iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torchvision/ops/boxes.py", line 36, in nms
    return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
  File "/home/ubuntu/.conda/envs/efficientdetpytorch/lib/python3.6/site-packages/torch/_ops.py", line 61, in __getattr__
    op = torch._C._jit_get_operation(qualified_op_name)
RuntimeError: No such operator torchvision::nms

解决：

pip install torch==1.3.0

pip install torchvision==0.4.1

若无法解决欢迎留言，我看到回复。