首先查看是否使用了import cv2
如果有import cv2,说明是opencv的问题
因为如果你的opencv是本地编译的,那么很可能使用了cudnn进行编译,那么这个cv2就会占用显存,并且和tensorflow冲突。
解决办法:重新编译opencv: http://www.cnblogs.com/huangshiyu13/p/6647375.html