如何在 Windows 10 上轻松安装 Detectron2
在 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
布施恩德可便相知重
微信扫一扫打赏
支付宝扫一扫打赏