系统相关
首页 > 系统相关> > 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,https,python,detectron2,pytorch,install,pip,安装
来源: https://blog.csdn.net/qq_24505417/article/details/118944032