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

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐