首先查看是否使用了import cv2
如果有import cv2,说明是opencv的问题
因为如果你的opencv是本地编译的,那么很可能使用了cudnn进行编译,那么这个cv2就会占用显存,并且和tensorflow冲突。
解决办法:重新编译opencv
黄世宇/Shiyu Huang's Personal Page:https://huangshiyu13.github.io/
微信扫一扫
首先查看是否使用了import cv2
如果有import cv2,说明是opencv的问题
因为如果你的opencv是本地编译的,那么很可能使用了cudnn进行编译,那么这个cv2就会占用显存,并且和tensorflow冲突。
解决办法:重新编译opencv
黄世宇/Shiyu Huang's Personal Page:https://huangshiyu13.github.io/
相关推荐