GiRay
/
detectron
Not watched
1
0
Fork 0
Code
Releases 0 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
detectron/INSTALL.md

3.0 kB
Raw Permalink Blame History

Installation

Our Colab Notebook
has step-by-step instructions that install detectron2.
The Dockerfile
also installs detectron2 with a few simple commands.

Requirements

  • Linux or macOS
  • Python >= 3.6
  • PyTorch 1.3
  • torchvision that matches the PyTorch installation.
    You can install them together at pytorch.org to make sure of this.
  • OpenCV, needed by demo and visualization
  • fvcore: pip install git+https://github.com/facebookresearch/fvcore
  • pycocotools: pip install cython; pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
  • VS2019/CUDA10.1

Build Detectron2

After having the above dependencies, run:

git clone https://github.com/facebookresearch/detectron2.git
cd detectron2
python setup.py build develop

# or if you are on macOS
# MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py build develop

# or, as an alternative to `setup.py`, do
# pip install [--editable] .

Note: you may need to rebuild detectron2 after reinstalling a different build of PyTorch.

Common Installation Issues

  • Undefined torch/aten symbols, or segmentation fault immediately when running the library.
    This may be caused by the following reasons:

    • detectron2 or torchvision is not compiled with the version of PyTorch you're running.

      If you use a pre-built torchvision, uninstall torchvision & pytorch, and reinstall them
      following pytorch.org.
      If you manually build detectron2 or torchvision, remove the files you built (build/, **/*.so)
      and rebuild them.

    • detectron2 or torchvision is not compiled using gcc >= 4.9.

      You'll see a warning message during compilation in this case. Please remove the files you build,
      and rebuild them.
      Technically, you need the identical compiler that's used to build pytorch to guarantee
      compatibility. But in practice, gcc >= 4.9 should work OK.

  • Undefined cuda symbols. The version of NVCC you use to build detectron2 or torchvision does
    not match the version of cuda you are running with.
    This happens sometimes when using anaconda.

  • "Not compiled with GPU support": make sure

    python -c 'import torch; from torch.utils.cpp_extension import CUDA_HOME; print(torch.cuda.is_available(), CUDA_HOME)'

    int valid outputs at the time you build detectron2.

  • "invalid device function" or "no kernel image is available for execution": two possibilities:

    • You build detectron2 with one version of CUDA but run it with a different version.
    • Detectron2 is not built with the correct compute compability for the GPU model.
      The compute compability defaults to match the GPU found on the machine during building,
      and can be controlled by TORCH_CUDA_ARCH_LIST environment variable during installation.