首先查看是否使用了import cv2

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

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

解决办法：重新编译opencv

黄世宇/Shiyu Huang's Personal Page：​​https://huangshiyu13.github.io/​​