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 #include #include #include #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 #inc..