ubuntu16.04下detectron2 安装

1. 基础环境:

显卡:Nvidia titan xp

驱动:460.73.01

cuda: 9.0

2. 安装:

follow installation tutorials: https://detectron2.readthedocs.io/en/latest/tutorials/install.html

NOTE:

(1) create envs:

conda create -n detectron2 python=3.6.2

source activate detectron2

(2) Install pytorch

torchvision版本需要和pytorch的版本相适应,可参考 pytorch.org 安装.

conda install pytorch==1.7 cudatoolkit=10.1 torchvision=0.8.2 torchaudio==0.7.2 -c pytorch

(3) opencv

pip install opencv-python

(4)detectron2 

git clone https://github.com/facebookresearch/detectron2.git

python -m pip install -e detectron2

(5)

rm -rf build/ **/*.so

(6) rebuild: 

python -m pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.6/index.html
上一篇:Detectron2 获取网络中间层结果的途径


下一篇:深度学习目标检测框架 detectron2 安装流程