首先查看是否使用了import cv2

如果有import cv2,说明是opencv的问题

因为如果你的opencv是本地编译的,那么很可能使用了cudnn进行编译,那么这个cv2就会占用显存,并且和tensorflow冲突。

 

解决办法:重新编译opencv: http://www.cnblogs.com/huangshiyu13/p/6647375.html

 

相关文章: