在 Windows 上安装 Detectron2 并不是一件容易的事，因为正式版的 Detectron2 在这个平台上是官方不支持的，而且 Windows Subsystem for Linux 也无法访问机器的 GPU。按照接下来的步骤，您可以快速安装和使用最新版本的Detectron2，完美运行。

第 1 步：创建 conda 环境

此步骤不是必需的。您可以使用下面描述的两个命令行创建一个空环境来测试 detectron2：https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an -环境与命令

conda create -n myenv python=3.7
conda activate myenv

将 myenv 替换为您的环境名称。

第 2 步：安装 Cuda

转到 Nvidia 的官方网站并下载 cuda Tookit 可执行文件。我推荐 Cuda 10.1，它可以与火炬版本 1.6、1.5 和 1.4 一起使用。
https://developer.nvidia.com/cuda-10.1-download-archive-base

然后，像安装任何程序一样安装 Cuda 工具包。

第 3 步：安装手电筒

转到https://pytorch.org/get-started/locally/并选择PyTorch 1.6。PyTorch 1.7 及更高版本不工作。如果你已经安装了cuda 10.1 或更高版本，命令应该是：

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

注意：PyTorch 1.6 不支持 cudatoolkit=11。但是如果你已经安装了 CUDA 11.x，cudatoolkit=10.1 仍然可以正常工作。

第 4 步：安装 Cython 和 Pycocotools

在安装 Pycocotools 之前，你需要 Cython。

pip install cython

然后使用下面的存储库，它可以很好地与 Windows 一起安装 Pycocotools。

pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"

第 5 步：安装 Detectron2

现在，您已准备好安装 Detectron2。克隆 Detectron2 的以下存储库。

git clone https://github.com/DGMaxime/detectron2-windows.git

此存储库已准备好与 Windows 一起使用。一些文件已被修改。您可以在此处找到这些文件：

setup.py
detectron2/engine/ defaults.py
detectron2/layers/csrc/cocoeval/ cocoeval.cpp

克隆完成后，安装它。

cd detectron2-windows
pip install -e .

如果安装成功，您应该会看到消息“Successfully installed detectron2”。

第六步：测试Detectron2的安装

在同一个存储库中， tests文件夹中有一个测试文件。运行名为test_windows_install.py的文件。您必须先安装 opencv 才能执行测试。

pip 安装 opencv-python 
python 测试/test_windows_install.py

结果：

希望这个故事对你有帮助！

文章来源：https://dgmaxime.medium.com/how-to-easily-install-detectron2-on-windows-10-39186139101c