在多个进程中使用tensorflow-gpu

时间:2018-07-23 16:34:48

标签: python tensorflow pytest

我正在尝试为使用tensroflow-gpu的应用程序编写测试。我正在努力使用pytest-xdist和-n选项并行运行测试。为了减少Tensorflow正在使用的GPU内存量并在可用的GPU之间分布测试,我正在使用tf.GPUOptions,但这似乎还不够。 Tensroflow因内部错误而崩溃。

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction,
                            visible_device_list=gpu_for_process,
                            allow_growth=False)

是否可以在多个进程中使用tensorflow-gpu?至少出于测试目的。

0 个答案:

没有答案