该模型不适合 Tensorflow 和 GPU 的 Keras 错误答案

【问题标题】：The model doen't fit withTensorflow and Keras Error with GPU该模型不适合 Tensorflow 和 GPU 的 Keras 错误
【发布时间】：2021-05-08 18:51:51
【问题描述】：

我使用 tensorflow-gpu 2.4.0 安装了 cuda 11.0 和 cudnn 8.0。当我拟合模型时，我得到了这个错误。

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.
     [[node sequential/conv2d/Relu (defined at <ipython-input-23-397fbbe99754>:1) ]] [Op:__inference_train_function_1671]

Function call stack:
train_function

我尝试安装 keras-gpu，但是 all 环境和 tensorflow 都不起作用。

【问题讨论】：

在 CLI 上输入 nvidia-smi 并验证您的 cuda 版本。然后确保您通过 conda 安装了正确的版本，例如 conda install cudatoolkit=10.1
NVIDIA-SMI 461.40 Driver Version: 461.40 CUDA Version: 11.2 from tensorflow versions 我用python3.8这样安装conda install cudatoolkit=11.0 cudnn=8.0 -c=conda-forge
你看到那里的不协调了吗？您有 CUDA 11.2，但工具包上只安装了 11.0。
如何替换我在代码conda install cudatoolkit=11.0 cudnn=8.0 -c=conda-forge中写的安装cuda 11.0
conda install cudatoolkit=11.2 cudnn=8.0 -c=conda-forge

标签： python tensorflow keras gpu

【解决方案1】：

import tensorflow as tf
gpu = tf.config.experimental.list_physical_devices('GPU')
print("Num GPUs Available: ", len(gpu))
tf.config.experimental.set_memory_growth(gpu[0], True)

这解决了错误。

【讨论】：