win10 编译安装detectron2

本次安装环境
cuda 10.1 下载地址
pytorch 1.5.1 命令: pip install torch1.5.1+cu101 torchvision0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html 更多安装方式
vs 2017 下载地址
fvcore 0.1.2

下载源码

修改 detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu
1.注释块/* #ifdef WITH_CUDA…#endif*/
2.在后面添加一句 #include “box_iou_rotated/box_iou_rotated_utils.h”

// 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"

编译

进入下载代码的目录执行

""" 
@author: qq群686070107
""" 
python -m pip install -e detectron2(或者detectron2-master)
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 技术工厂 设计师:CSDN官方博客 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值