代碼下載自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文件即可通過。
編譯成功後: