Нет такого оператора torchvision :: nms

Question

Когда я пытаюсь запустить обнаружение yoloV3, возникает ошибка

op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms

Хотя этот код является исходным кодом torchvision, я пытаюсь несколько раз исправить код подсказками с ошибкой.

Answer 1

Поскольку я столкнулся с множеством препятствий для установки torch и torchvison, я не отказываюсь переустанавливать enviroment.Running 'conda list' версия torch и torchvison, я обнаружил, что они не являются несовместимыми, версии двух пакетов установлены являются:

 torchvision-0.5.0+cu92-cp37-cp37m-win_amd64.whl
 torch-1.4.0+cpu-cp37-cp37m-win_amd64.whl

Я изменяю версию torchvison на

 torchvision-0.5.0+cpu-cp37-cp37m-win_amd64.whl

Я имел дело с моей проблемой до сих пор. Примечание: окружение моей ОС и python общаются:

  Windows_X64
  python3.7