首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >在Windows 10上安装Detectron2

在Windows 10上安装Detectron2
EN

Stack Overflow用户
提问于 2020-03-11 08:31:04
回答 7查看 17.8K关注 0票数 8

我试着安装Facebook的Detectron2和这个官方回购。在此基础上,detectron2只能在linux上安装。但是,我正在Windows操作符上运行一个服务器。有人知道如何在Windows上安装吗?

EN

回答 7

Stack Overflow用户

发布于 2021-10-28 07:33:35

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

这些步骤在我的RTX 3070上对我有效。

  1. 安装Anaconda https://docs.anaconda.com/anaconda/install/windows/
  2. 创建一个包含以下代码的environment.yml文件。
代码语言:javascript
复制
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终端中运行您的脚本,这是没有问题的,但是在从VS代码运行脚本时,我也有这个问题ImportError: DLL load failed: The specified module could not be found.,所以如果您碰巧有这个问题,我通过从anaconda终端卸载和重新安装问题模块来修复它。

代码语言:javascript
复制
pip uninstall numpy
pip install numpy
票数 8
EN

Stack Overflow用户

发布于 2021-12-27 16:39:58

在Windows中安装detectron2有点棘手。我挣扎了整整一周才让它运转起来。为此,我创建了一个新的anaconda环境(以配合detectron2对py手电筒和torchvision的版本要求),并从在该环境中安装cudatoolkit和cudnn开始。这可能是最好的方式,以避免混乱现有的水蟒环境。

下面是一步一步的过程(用我的笔记本电脑与Windows 10和RTX2070 GPU进行验证):

  1. 创建一个anaconda环境(比如'detectron_env'): (注:python 3.8不起作用,3.7起作用 conda创建-n detectron_env python=3.7
  2. 激活detectron_env: conda激活detectron_env
  3. 安装cudatoolkit: (注:cuda版本号应该与安装在计算机中的版本号相匹配(在我的例子中是11.3 )。您可以通过在anaconda提示符窗口中键入"nvcc -V“进行检查。有关更多信息,请参见https://pytorch.org/) 安装-c anaconda cudatoolkit=11.3
  4. 安装cudnn: (注:不要指定版本号。它将被自动计算出来) 安装-c anaconda cudnn
  5. 安装pywin32: 安装-c anaconda pywin32
  6. 安装火把、火炬和torchaudio: (注:cudatoolkit的版本号应与步骤3中的版本号相匹配。detectron2要求的版本号等于或高于1.8的版本将自动安装。 conda安装手电筒cudatoolkit=11.3 -c手电筒
  7. 检查是否启用了GPU,以便为py手电筒: 启用python,导入torch并键入'torch.cuda.is_available()‘

你应该得到“真”。但是,如果您发现GPU没有为py手电筒启用,请转到步骤1),然后用不同的版本号重试cuda和/或python。

  1. 安装一些软件包: (注:你应该安装忍者。否则,建立和构建过程将不会顺利进行)
代码语言:javascript
复制
- conda install -c anaconda cython
代码语言:javascript
复制
- pip install opencv-python
代码语言:javascript
复制
- pip install git+https://github.com/facebookresearch/fvcore
代码语言:javascript
复制
- pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
代码语言:javascript
复制
- pip install av
代码语言:javascript
复制
- conda install -c anaconda scipy
代码语言:javascript
复制
- conda install -c anaconda ninja
  1. 转到要安装detectron2的目录。
  2. Git克隆以下存储库: (注:detectron2的文件夹名应该与“探测2”不同。在我的情况下,我用了“探测仪”。否则,火把的路径将被混淆) git克隆https://github.com/facebookresearch/detectron2.git detectron_repo
  3. 安装依赖关系: (注:不要输入克隆的detectron_repo目录) pip安装-q -e detectron_repo
  4. 转到detectron_repo目录: cd detectron_repo
  5. 构建detectron2: python开发

如果上面的操作不成功,您可能需要从头开始重新启动,或者重新安装pytorch。如果重新安装pytocrh,则需要重新构建detectron2。

如果上述方法成功,那么

  1. 测试: 转到演示/目录,通过对任何图像(例如.jpg)的输入路径进行配置,运行以下脚本: python ../configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --config-file demo.py --输入 -opts MODEL.WEIGHTS
票数 5
EN

Stack Overflow用户

发布于 2021-07-28 14:41:37

这是安装说明。要安装( Detectron2的最新版本),请执行以下命令。

  • Windows10
  • Python = 3.7.9
  • 比托尔= 1.7.1
  • 火炬视野= 0.8.2
  • 库达= 11.0
  • detectron2 = 0.5

在航站楼:

代码语言:javascript
复制
python -m pip install git+https://github.com/facebookresearch/detectron2.git
票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/60631933

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档