ubuntu 16.04 安装 detectron2

nvcc -V 查看 CUDA 版本号，这里以 python 3.6， CUDA 10.2 为例：

一、搭建环境

1. 创建一个虚拟环境，比如起名为 dt2 或其他

conda create -n dt2 python=3.6
conda activate dt2

2. 离线安装 pytorch（找与 CUDA python 匹配的版本）

直接用 conda 提供的语句在线安装可能报错

torch 和 torchvision 的下载地址

INSTALLING PREVIOUS VERSIONS OF PYTORCH

我下载的版本为：

• torch-1.6.0-cp36-cp36m-linux_x86_64.whl
• torchvision-0.6.0-cp36-cp36m-linux_x86_64.whl

下载好两个文件，使用 pip install xxx.whl 安装两个包即可

3. 安装 opencv

pip install opencv-python

4. 安装 ninja

conda install -c conda-forge ninja

5. 下载编译 detectron2

git clone https://github.com/facebookresearch/detectron2.git
python -m pip install -e detectron2

---

二、测试

在当前环境下输入 python，看是否能成功导入下面这些库：

# Setup detectron2 logger
import detectron2
from detectron2.utils.logger import setup_logger
setup_logger()

# import some common detectron2 utilities
from detectron2 import model_zoo
from detectron2.engine import DefaultPredictor
from detectron2.config import get_cfg
from detectron2.utils.visualizer import Visualizer
from detectron2.data import MetadataCatalog, DatasetCatalog