tensorflow2.0卷积报错:Failed to get

小白踩坑，记录一下。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'