win10中安裝mmcv和mmdetection的問題



目前環境:vs2019preview-2.0(14.28.29617+python3 32位 64位),cuda10.2,torch1.6.0,torchvision0.7.0,mmcv-1.1.5,mmdetection-2.7.0

 

新版本里編譯安裝在我的電腦上不可行,以下跳過,直接看第0. 步

先命令行里運行一下ninja -v,然后添加到防火牆的白名單中。

 

由於cl的編碼問題,需要修改:D:/ProgramData/Anaconda3/envs/mmlabv2/Lib/site-packages/torch/utils/cpp_extension.py中的

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode().strip())

改為(注意有空格)

match = re.search(r'(\d+)\.(\d+)\.(\d+)', compiler_info.decode(" gbk").strip())

(如果還找不到cl的話,建議在VS的安裝目錄內搜索cl.exe,然后把目錄,添加到環境變量PATH內,mmcv編譯時使用x86內的那套x64,

PATH新增內容形如C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29617\bin\Hostx86\x64 (先行版用戶的文件夾可能不是Community而是Preview),每次更新后注意VS版本號變化)

 

安裝過程:

0. 注意要使用Anaconda PowerShell prompt而不是CMD或者VS/VSCode內的命令行。編譯安裝可以嘗試刪除線內的內容輸入以下信息進行變量預設 6.1是顯卡算力,視具體顯卡而定

$env:TORCH_CUDA_ARCH_LIST="6.1"
$env:MMCV_WITH_OPS = 1
$env:MAX_JOBS = 4

1. mmcv在github中tag到1.2.1,在解壓后的目錄內,python setup.py build_ext 然后 python setup.py develop

直接安裝預編譯文件,在https://download.openmmlab.com/mmcv/dist/index.html中下載./1.1.5/torch1.6.0/cu102/mmcv_full-1.1.5+torch1.6.0+cu102-cp37-cp37m-win_amd64.whl

使用pip本地安裝這個文件

 

2. cocoapi中進入到PythonAPI目錄, python setup.py install

 

3. mmdetection在github中tag到2.7.0,解壓后的目錄內,pip install -r requirements/build.txt 然后 python setup.py develop

 

 

 

參考:https://mmcv.readthedocs.io/en/latest/build.html


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM