detectron2 install error

 

pip install -e .

detectron2 install 중 error 발생.

 

 

detectron2/layers/csrc/nms_rotated/nms_rotated_cuda.cu

 

를 열어 아래와 같이 수정 후 다시 install 하면 해결 된다.

 

<수정 전>

// 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.
#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
#include "box_iou_rotated/box_iou_rotated_utils.h"