如何解决：UnknownError：图形执行错误：

Question

我有2.14版本的keras和tensorflow。我的cuda版本是：

我正在尝试拟合一个简单的模型，但出现以下异常：

UnknownError: Graph execution error:

代码是：

Problem_with_Graph_execution_error.py

我的电脑是 ubuntu 20.04，带有 python 3.10.14 (ipython 8.27.0)。

请任何可以帮助我的想法或经验。

Answer 1

尝试添加以下代码行：

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], enable=True)