gpt4 book ai didi

pytorch - 没有这样的运算符(operator) torchvision::nms

转载 作者:行者123 更新时间:2023-12-03 19:17:08 30 4
gpt4 key购买 nike

当我尝试运行 yoloV3 检测时,发生了错误

op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torchvision::nms

虽然这段代码是 torchvision 的源代码,但我尝试了一些时间来通过失败的提示来纠正代码。

最佳答案

由于我在安装 torch 和 torchvison 时遇到了很多障碍,我不情愿重新安装环境。运行 '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

到目前为止,我已经处理了我的问题。
注:我的os和python的环境是相通的:
  Windows_X64
python3.7

关于pytorch - 没有这样的运算符(operator) torchvision::nms,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/60247432/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com