代码下载自2020-06-17更新代码。若代码结构不一致变化较大时,可对代码回滚至当前版本:
git reset --hard ae931b4
1、系统环境
win10、vs2017、cuda10.1、pytorch1.4、torchvision0.5。
2、配置conda虚拟环境
conda create --n detectron2 python=3.6
conda activate detectron2
conda install pytorch=1.4 torchvision=0.5 cudatoolkit=10.1
conda install pytorch-nightly ipython
# dependencies
pip install ninja yacs cython matplotlib==2.2.4 tqdm opencv-python
python setup.py build develop
编译后,缺少包,进行安装:
pip install fvcore tensorboard future pydot mock
conda install pillow # PIL包,windows上需要编译,通过conda安装
重新编译,通过:
python setup.py build develop
注:6月4号后,版本若cuda编译失败,那是因为AMD公司提交的一条记录,修改cuda编译引起的。代码回滚值前一条记录,替换对应csrc文件即可通过。
编译成功后:
detectron2配置安装,总目录: