1. apex 安装报错 Cuda extensions are being compiled with a version of Cuda that does not match the verson
看这个意思是说cuda跟pytorch的版本不对应。使用下面命令安装,成功。
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir ./
2. ImportError: cannot import name 'cython_nms' from 'maskrcnn_benchmark.layers.cython_nms' (unknown location)
简单粗暴的解决方式,我看代码里面没有用到cython_nms,所以我把这句注释掉了,至于报错的原因,我都重装了环境,不过就python版本不同,不知道是不是这个原因,
# from maskrcnn_benchmark.layers.cython_nms import cython_nms
第二个解决方法
首先下载 setuptools: conda install setuptools
然后移到cython_nms目录下
: cd maskrcnn_benckmark/layers/cython_nms
最后运行命令: python setup.py develop