tensorflow 1.4 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 错误的解决办法

2023年11月21日 974次阅读 来源: Oliver_5fb2

tensorflow 1.4  cuda8.0 cudnn 6.0

执行gpu版本的 MNIst 时候报错：

E tensorflow/stream_executor/cuda/cuda_dnn.cc:371] could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 

E tensorflow/stream_executor/cuda/cuda_dnn.cc:338] could not destroy cudnn handle: CUDNN_STATUS_BAD_PARAM

F

tensorflow/core/kernels/conv_ops.cc:672] Check failed:

stream->parent()->GetConvolveAlgorithms(

conv_parameters.ShouldIncludeWinogradNonfusedAlgo(),

&algorithms) 

试了很多办法

（1）：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))这个方法对我来说不管用，还是之前的错误；

(2):执行sudo rm -f ~/.nv 解决问题

参考：https://github.com/tensorflow/tensorflow/issues/6698

    原文作者：Oliver_5fb2
    原文地址: https://www.jianshu.com/p/8be286b38027
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞