CPU版本的pytorch maskrcnn怎么运行demo以及训练自己的shu'ju

Question

最近刚学习深度学习，准备跑下pytorch版本的maskrcnn，但是因为电脑显卡是a卡，所以照着github（https://github.com/facebookresearch/maskrcnn-benchmark）上的教程安装了必要组件，但是没有装cuda和apex，结果运行demo（命令：python webcam.py --min-image-size 300 MODEL.DEVICE cpu）的shi'hou总是出现以下提醒没有装apex，可是apex要求环境是cuda，我没法装cuda啊，有没有大神来解答下，万分感谢！！！

 File "h:\mask-rcnn\maskrcnn-benchmark-windows10installationfixes_ver2\maskrcnn_benchmark\layers\nms.py", line 5, in <module>
    from apex import amp
ModuleNotFoundError: No module named 'apex'

Answer 1

我之前也遇到了这个问题，尝试后是这样解决的：1，安装torch和torchvision的时候选的cuda版本；2，按要求安装apex组件，安装命令如下：python setup.py install，注意千万别加后面的--cuda_ext --cpp_ext指令，否则会报错。这样就正常安装好啦~调试了一下，真香~新人首次答题，给个赞哦~