美文网首页我爱编程
tensorflow 1.4 could not create

tensorflow 1.4 could not create

作者: Oliver_5fb2 | 来源:发表于2018-03-06 22:11 被阅读0次

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

相关文章

网友评论

    本文标题:tensorflow 1.4 could not create

    本文链接:https://www.haomeiwen.com/subject/qkqdfftx.html