Tensorflow Cuda错误未能获得设备0的设备属性13

Question

使用tensorflow-keras训练神经网络时，我面临的问题。我得到了这个错误：

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

我一开始就发现了这个错误。然后，我引用了failed-to-get-device-attribute-13-for-device-0中的解决方案。我更新了图形驱动程序。这对3-4的运行有效，现在我又得到了同样的错误。

以下是我所处环境的详细情况：

(Anaconda)

• Tensorflow 2.1

• Nvidia GeForce RTX 2060，6GB Graphics

• Windows 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)