Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

1190 단어 TensorFlowKeras
keras yolo v3 실행 Could not create cudnn handle: CUDNNSTATUS_ALLOC_FAILED 문제, 코드의 맨 처음에 다음 코드를 추가합니다.
import tensorflow as tf
from keras import backend as K
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
K.set_session(sess)

직접 테스트가 유용하다. 원인은 GPU 점용 문제로 인한 것일 수 있다.참조:https://blog.csdn.net/weixin_42769131/article/details/88848478

좋은 웹페이지 즐겨찾기