【已解决】detectron2安装失败： error: identifier “single_box_iou_rotated“ is undefined

最新推荐文章于 2025-10-22 17:36:18 发布

原创最新推荐文章于 2025-10-22 17:36:18 发布 · 1.3k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#python #深度学习 #object detection

部署运行你感兴趣的模型镜像

1、问题描述

当我在detectron2的Github上把安装包下载到本地后，我在Anaconda中通过以下命令对detectron2进行安装：

python -m pip install -e path_to_detectron2_Directory

但是会安装失败并有以下报错：

2、解决办法

修改源码"detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu":

修改前：

// Copyright (c) Facebook, Inc. and its affiliates.
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif

修改后：

// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>
#include <c10/cuda/CUDAGuard.h>
#include <ATen/cuda/CUDAApplyUtils.cuh>
//NOTE: replace relative import
/*#ifdef WITH_CUDA
#include "../box_iou_rotated/box_iou_rotated_utils.h"
#endif
// TODO avoid this when pytorch supports "same directory" hipification
#ifdef WITH_HIP
#include "box_iou_rotated/box_iou_rotated_utils.h"
#endif*/
#include "box_iou_rotated/box_iou_rotated_utils.h"