当前版本的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