我正在尝试为使用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?至少出于测试目的。