Detectron2 installation method

install detectron2

install detectron2 Methods , The official tutorial ：Installation — detectron2 0.6 documentation

The first method ：Build Detectron2 from Source

Official source code ：

python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
# (add --user if you don't have permission)

# Or, to install it from a local clone:
git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2

# On macOS, you may need to prepend the above commands with a few environment variables:
CC=clang CXX=clang++ ARCHFLAGS="-arch x86_64" python -m pip install ...

Be careful ： If you download first detectron2, Method of reinstallation , Do not download after   cd detectron2, Otherwise, there will be no “step.py” Error of .

My environment is CUDA：11.6, pytorch：1.10.2

Install in the first way , There will be CUDA And torch Version mismatch . An error is as follows ：

The second method ：Install Pre-Built Detectron2 (Linux only)

Know your own CUDA Version and torch After version , Using the second method is faster , And it's not easy to make mistakes , I installed it in this way .

My environment is CUDA：11.6, pytorch：1.10.2, I chose CUDA==10.2, torch==1.10.

  success ！