1. 程式人生 > 實用技巧 >windows10上安裝mmdetection2.5.0 + pytorch1.6 + CUDA10.1 (python==3.6)

windows10上安裝mmdetection2.5.0 + pytorch1.6 + CUDA10.1 (python==3.6)

之前只能在windows上安裝py37版本的, 目前已支援py36了.(2020-10-19)
能不能安裝, 主要是看能裝什麼版本的mmcv-full.
在mmcv的安裝包網址上可以看到, 目前已經有mmcv_full-1.1.5+torch1.6.0+cu101-cp36-cp36m-win_amd64.whl的安裝包了.
而在mmdetection的官方安裝教程中可以看到, mmdetection>=2.3.0版本是需要mmcv-full>=1.0.2的.
能夠直接用pip安裝mmcv-full, 這樣一來, 安裝就很簡單了.



以下是教程:

0. conda create -n mmdetection python=3.6

1. 安裝pytorch
conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch

2.安裝mmcv-full
pip install mmcv-full==1.1.5+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html
或者
pip install mmcv-full==latest+torch1.6.0+cu101 -f https://download.openmmlab.com/mmcv/dist/index.html

3. pip install Cython

4. 下載原始碼
從官方git release下載: https://codeload.github.com/open-mmlab/mmdetection/zip/v2.5.0
或者git clone https://github.com/open-mmlab/mmdetection.git (截止2020-10-19, 最新版本是2.5.0)
cd mmdetection

5. 安裝必須包
cd mmdetection
pip install -r requirements/build.txt

6. 編譯mmdetection原始碼
python setup.py install develop (這個安裝後,會把程式碼複製到python site-packages下面)

or
python setup.py build develop (推薦用這個,安裝後使用的是所下載的mmdetection目錄的程式碼, 方便修改

7. 測試
python
import mmdet
import mmdet.core
print(mmdet.__version__)