如何在 Windows 上安装 Detectron2

Detectron2 是由 Facebook AI Research 构建的一款功能强大的开源物体检测和分割框架。它广泛应用于计算机视觉应用的研究和开发。然而，由于各种依赖项，在 Windows 11 上安装 Detectron2 可能会有些棘手。在本指南中，我将逐步指导您在 Windows 11 系统上设置 Detectron2。

步骤 1：设置 Conda 环境

首先，让我们创建一个新的 conda 环境来隔离安装：

conda create --name detectron2_env python=3.11.7
conda activate detectron2_env

第 2 步：安装 CUDA

如果您拥有 NVIDIA GPU，则需要安装 CUDA 工具包。请确保您拥有与您的 GPU 兼容的正确版本：

conda install cuda=12.1 -c nvidia

步骤3：安装PyTorch

使用所需的 CUDA 版本安装 PyTorch：

conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=12.1 -c pytorch -c nvidia

对于其他版本的 CUDA，以及所需的 PyTorch 版本，可以参考https://pytorch.org/get-started/locally/。

如果你使用 CPU，请使用

conda install pytorch torchvision torchaudio cpuonly -c pytorch

步骤 4：更新 Visual C++ Redistributable

确保您的系统已安装最新的 Visual C++ Redistributable。您可以从 Microsoft 官方网站下载并安装。

步骤5：安装Cython和COCO工具

Cython 是构建 Detectron2 的先决条件，而评估所需的 COCO 工具：

pip install cython
conda install conda-forge::pycocotools

步骤6：克隆Detectron2存储库

从 GitHub 克隆 Detectron2 存储库：

git clone https://github.com/facebookresearch/detectron2.git

步骤7：安装Detectron2

python -m pip install -e detectron2

就这样！您已成功在 Windows 11 系统上安装 Detectron2。现在您可以开始使用它执行各种计算机视觉任务，例如对象检测、实例分割等。

无论何时想要使用 Detectron2，请记得激活您的 conda 环境。

编码愉快！🤗🤗