win10下detectron2環境配置

代碼下載自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文件即可通過。

編譯成功後:

 

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章