如何使用C++ API限制张量流（r1.1）中GPU的使用情况

Question

我想限制C++ API中张量流的GPU分配。我知道如何在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);