如何使用C++ API限制tensorflow(r1.1)中的GPU使用率

Question

我想在C++ API中限制GPU分配的tensorflow.我知道如何在python中做到这一点,它可能与传递给NewSession()的struct SessionOptions有关,但是我找不到更多关于如何做的具体信息.

编辑:Stackoverflow表明我的问题是重复的.但是我已经链接到其他帖子了.这是如何在python中做到这一点.我特别想问如何用C++做.

Answer 1

结果很简单:

tensorflow::SessionOptions session_options;
session_options.config.mutable_gpu_options()->set_allow_growth(allow_growth);
session_options.config.mutable_gpu_options()->set_per_process_gpu_memory_fraction(per_process_gpu_memory_fraction);