trouble shooting
no module named cv2 :
pip install opencv-python
No module named 'object_detection'
pip install tensorflow-object-detection-api
Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
GPU显存设置有问题,需要设置为仅在需要时申请显存。
import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options=tf.compat.v1.GPUOptions(allow_growth=True))
sess = tf.compat.v1.Session(config=config)
Last updated