在windows10下部署MaskRCNN成功,但是为了后续自己训练模型,将tensorflow卸载,安装了对应的CUDA,cudnn以及tensorflow-gpu,然而在运行demo.ipynb的时候没有像之前用CPU版本的那样正常显示标记后的图片,而且notebook一直在运行的过程中,如下图所示:
命令行中信息如下:
在windows10下部署MaskRCNN成功,但是为了后续自己训练模型,将tensorflow卸载,安装了对应的CUDA,cudnn以及tensorflow-gpu,然而在运行demo.ipynb的时候没有像之前用CPU版本的那样正常显示标记后的图片,而且notebook一直在运行的过程中,如下图所示:
命令行中信息如下:
GPU显存不够吗?你可以用cpu推理下,如果cpu推理没问题的话就是gpu现存不够,图片1024*1024太大了,改下输入图片大小或者batch size改成1