【第二次安装】安装detectron2 ——0.6版本。

系列文章目录

上一次安装detectron2 的0.5版本后，打算使用0.6的版本。

---

---

前言

最新的模型使用0.6版本的detectron2，pip卸载0.5版本。

一、下载0.6版本的detectron2

git下载到目标文件夹，没有则创建同名目录。

git clone https://github.com/facebookresearch/detectron2.git  目标文件夹

使用detectron2-0.6作为目标文件夹。

二、修改其中的一句

1.修改配置文件

文件目录：detectron2-0.6\detectron2\layers\csrc\nms_rotated\nms_rotated_cuda.cu
第10行 添加#define WITH_HIP，如下所示。

2.编译安装detectron2

python -m pip install -e detectron2-0.6

正常安装。

三、参考资料

1. 在Windows11上安装Detectron2
2. 从零开始安装Detectron2

---

总结

安装detectron0.6，只要安装了c++编译器，基本不会有问题。