mmdetection安裝問題(nms is not compiled with GPU support)


在按照官方的安裝教程進行mmdetection安裝的時候,出現了一些問題。我的環境信息如下:
最后報錯信息如下:
從MMCV Compiler這兒來看,可以看見MMCV CUDA Compiler: not available。

原因是,我使用的mmcv-full的腳本是:

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.1/index.html

使用如下腳本安裝就可以了:

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html

原因在官網上可以找到:

原因是mmcv-full只在pytorch 1.x.0編譯過。

裝了好久,哭哭。


免責聲明!

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



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