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