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"
相关链接: