當前版本的maskrcnn嚴重依賴pythorch_nightly 1.0.0, 安裝過程中一不留神會由於各個包之間不兼容導致no module named maskrcnn_benchmark.
注意以下三點即可:
1. 先安裝ipython
2. 再單獨安裝torchvision。 conda install torchvision
3. 最後本地安裝pytorch-nightly. 先去https://anaconda.org/pytorch/pytorch-nightly/files下載合適的版本,包括python版本、cuda版本和cudnn版本。下載後使用 conda install --use-local xxxx安裝包名字 進行安裝
剩下步驟按照官方說明即可。
運行時如果報 no module named xxx 則直接搜索如何安裝xxx