問題描述

2019-11-11 10:23:34.837541: E tensorflow/stream_executor/cuda/cuda_dnn.cc:353] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

解決方法

在代碼中添加以下代碼：

config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True)) sess = tf.Session(config=config)

我總共嘗試添加了好幾個位置，有的位置報錯，有的不報了：

問題原因

GPU重復調(diào)用

因為可能在另一個地方調(diào)用了tensorflow，導致該問題出現(xiàn)，我在jupyter notebook中調(diào)用了tensorflow，我將它關掉，一切又好了。沒再報錯。

參考文章1：yolov3檢測報錯Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

參考文章2：window+tensorflow+cuda+cudnn 出現(xiàn)‘CUDnn_STATUS_ALLOC_FAILED’的可能解決辦法

參考文章3：Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED問題

總結(jié)

以上是生活随笔為你收集整理的yunyang tensorflow-yolov3 启动evaluate.py报错：Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED原因的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。