win10下基于anaconda的detectron2安装

在Anaconda中创建虚拟环境

例如，我给创建的虚拟环境起名叫mPytorch，执行下面的命令以创建这个环境：

conda create -n mPytorch python=3.7
1

创建完成后，使用conda activate语句激活该环境。当前激活了哪个环境，运行conda install或者pip install安装的库就存在于哪个环境。激活mPytorch后，运行官方命令安装pytorch和cudatoolkit。

conda activate mPytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
1
2

安装cuda的C语言开发包

注意，上一节cudatoolkit安装了哪个版本，就要去英伟达官网下载与其版本对应的C语言cuda开发包，并安装。其下载安装参考该网站。具体下载地址。
下载完成后，双击该安装包，接受默认安装选项，直至安装完成。

安装MSVC编译器

若下载了cuda10.x版本，则应该安装VS2019；若下载了cuda11.x版本，则应该安装VS2022。
安装完成后，应该在开始菜单可以找到x64 Native Tools Command Prompt for VS 2019命令行入口：

下载并编译安装detectron2

虽然官方声称暂不支持windows下编译安装，但是亲测可行。

在GitHub下载detectron2的源码并解压，或者使用git clone命令将仓库复制到本地（如果你安装了Git for windows的话）。

源代码文件夹如图所示：

根据官方教程，要编译和安装detectron2，只需要在命令行执行一句命令：

python -m pip install -e .\detectron2-main\
1

其中-e后面是源码文件夹路径，可以是绝对路径或相对路径。
但是，windows下想要成功完成编译，需要注意以下几点：

1. 安装一系列依赖：

pip3 install easydict
conda install pyyaml
conda install cffi
conda install cython
conda install scipy
pip install pycocotools
1
2
3
4
5
6

3. 打开你的x64 Native Tools Command Prompt for VS 2019或x64 Native Tools Command Prompt for VS 2022，并输入conda activate mPytorch激活conda环境。
4. 设置临时环境变量

set DISTUTILS_USE_SDK=1
1

5. 修改源码文件nms_rotated_cuda.cu:
   
   打开nms_rotated_cuda.cu，将下图中框出部分在行首用’//'注释掉即可。这是为了解决错误信息nms_rotated_cuda.cu(68): error: identifier "single_box_iou_rotated" is undefined，也可以参考此博客。
   
6. 运行python -m pip install -e .\detectron2-main\进行编译和安装。

测试使用detectron2

需要安装opencv。

pip install opencv-python
1

根据官方示例，下载一个训练好的模型，model_final_280758.pkl，并把它放在detectron2源码的demo文件夹下。
然后在命令行进入demo文件夹，运行命令：

python .\demo.py --config-file ..\configs\COCO-Detection\faster_rcnn_R_50_FPN_3x.yaml --input 你的输入图片名  --opts MODEL.WEIGHTS model_final_280758.pkl
1