ubuntu16.04下detectron2 安装

本文主要是介绍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

这篇关于ubuntu16.04下detectron2 安装的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！