tensorflow2.0卷积报错:Failed to get

作者: 白小白db | 来源:发表于2019-10-17 09:11 被阅读0次

小白踩坑,记录一下。tensorflow2.0.0-beta1版本,cuda版本10.0.130,cudnn版本7.6.3.30。

在学习tensorflow2.0时,学习链接:简单粗暴TensorFlow2.0:使用 Keras 实现卷积神经网络

卷积神经网络(CNN)这部分的例子

替换以后运行报错:

UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]

解决办法:

添加下面两行

import os

os.environ['CUDA_VISIBLE_DEVICES'] = '/gpu:0'

相关文章

网友评论

    本文标题:tensorflow2.0卷积报错:Failed to get

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