1. 程式人生 > >could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

could not create cudnn handle:CUDNN_STATUS_INTERNAL_ERROR

在CPU下執行沒有問題,但是在GPU版本下,一直執行不成功。可以嘗試限制一下GPU的執行資源,新增如下程式碼:

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