一段时间运行的tensorflow-gpu有关于数组的MemoryError

时间:2019-03-09 13:46:40

标签: python tensorflow

当tensorflow-gpu运行一段时间(大约2hr〜)时,它有一个关于数组的MemoryError

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

我尝试:

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.4)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

CertCreateSelfSignCertificate()

但是,问题仍然存在 我不知道该如何解决错误。

GPU log file:

My python file:

计算机:MSI GE62笔记本

CPU:I7-6700HQ

GPU:GTX960M

Tensorflow:1.10.0

Python:3.5.6

CUDA9.0

我还有一个问题 当我使用tensorflowGPU时,它会在屏幕记录器应用程序运行时崩溃,对吧?

1 个答案:

答案 0 :(得分:0)

如果这是您的GPU内存给出错误,您将看到资源用尽,这似乎是python错误,而不是特定于tensorflow。由于代码效率低下,很可能是您的内存已满。参见this questio n。