在 Windows 10 上安装 Detectron2

Question

我尝试按照 这个官方 repo 安装 Facebook 的 Detectron2。在该 repo 之后，detectron2 只能安装在 linux 上。但是，我正在处理在 Windows 操作员上运行的服务器。有人知道如何在 Windows 上安装它吗？

原文由 Khiem Le 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

通过这个issue找到的答案： https ://github.com/facebookresearch/detectron2/issues/9

这些步骤适用于我的 RTX 3070。

1. 安装蟒蛇 https://docs.anaconda.com/anaconda/install/windows/
2. 创建一个包含以下代码的 environment.yml 文件。

 name: detectron2
channels:
  - pytorch
  - conda-forge
  - anaconda
  - defaults
dependencies:
  - python=3.8
  - numpy
  - pywin32
  - cudatoolkit=11.0
  - pytorch==1.7.1
  - torchvision
  - git
  - pip
  - pip:
    - git+https://github.com/facebookresearch/detectron2.git@v0.3

1. 启动 Anaconda 终端，导航到 yml 文件并运行 conda env create -f environment.yml

2. 激活环境 conda activate detectron2

你可以开始了。

编辑：如果你在 anaconda 终端中运行你的脚本，这没有问题，但我也有这个问题 ImportError: DLL load failed: The specified module could not be found. 从 VS Code 运行脚本时使用 numpy 和 Pillow 所以如果你碰巧遇到这个问题，我通过从 anaconda 终端中卸载并重新安装有问题的模块来修复它。

 pip uninstall numpy
pip install numpy

原文由 Austin Ulfers 发布，翻译遵循 CC BY-SA 4.0 许可协议