C/C++教程

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 安装的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!