python-3.x - Tensorflow Cuda 错误无法获取设备 0 的设备属性 13

Question

我在使用 tensorflow-keras 训练神经网络时遇到问题。我收到此错误：

F tensorflow/stream_executor/lib/statusor.cc:34] 尝试获取值而不是处理错误内部：无法获取设备 0 的设备属性 13：CUDA_ERROR_UNKNOWN：未知错误

我最初收到此错误。然后我参考了failed-to-get-device-attribute-13-for-device-0中的解决方案。我更新了图形驱动程序。这对一些 3-4 运行有效，现在我再次遇到相同的错误。

以下是我的环境的详细信息：

1. Python 3.7（蟒蛇）
2. 张量流 2.1
3. Nvidia GeForce RTX 2060，6GB 显卡
4. 视窗 10 版本 1809

Answer 1

那么在我的情况下，限制 GPU 内存是有效的。

在代码开头添加以下代码段：

import tensorflow as tf

gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)