最近刚学习深度学习,准备跑下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'
我之前也遇到了这个问题,尝试后是这样解决的:1,安装torch和torchvision的时候选的cuda版本;2,按要求安装apex组件,安装命令如下:python setup.py install,注意千万别加后面的--cuda_ext --cpp_ext指令,否则会报错。这样就正常安装好啦~调试了一下,真香~新人首次答题,给个赞哦~
https://blog.csdn.net/echineselearning/article/category/856221