当前位置:   article > 正文

Windows编译yolov5_obb的nms_rotated模块报错解决_importerror: /home/ps/v7obb/yolov7_obb-main/utils/

importerror: /home/ps/v7obb/yolov7_obb-main/utils/nms_rotated/nms_rotated_ex

一开始编译总共有34个bug,后面逐渐减少,记录期间的bug解决:

1、第一个bug

error: command 'c:\\program files\\nvidia gpu computing toolkit\\cuda\\v11.1\\bin\\nvcc.exe' failed with exit code 1
  • 1

这个可能是cuda装了好几个,删除不要用来编译的版本,去掉删除版本的环境变量。

2、第二个bug:

src/poly_nms_cuda.cu(236): error: declaration is incompatible with "void *memset(void *, int, size_t)"
  • 1

这个说的是代码信息无法识别不兼容,考虑到代码编辑格式,Linux代码可能用LF格式,Windows需要CRLF格式,找到报错的cu文件,把格式改为windows可识别的crlf格式,这去掉了一些报错。

3、第三个bug:

就是一个eps未定义,找了点办法,总算去掉了

e:/anaconda/envs/rolkechen2023/lib/site-packages/torch/include\c10/macros/macros.h(189): warning c4067: src/poly_nms_cuda.cu(63): error: identifier "eps" is undefined in device code
  • 1

找到的建议是:

const float eps = 1e-8;改为 (关键tmd有的却不用改)

#define eps 1e-8
  • 1
  • 2
  • 3

我在头文件下面加了
#define eps 1e-8
并把 const float eps = 1e-8; 注释了,应该可以直接换掉

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/369788
推荐阅读
相关标签
  

闽ICP备14008679号