如何使用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);