我想限制C ++ API中的Tensorflow的GPU分配。我知道如何在python中执行此操作，这可能与传递给NewSession（）的struct SessionOptions有关，但是我找不到有关执行此操作的更具体的信息。
编辑：Stackoverflow建议我的问题是重复的。但是，我已经链接到该其他帖子。那就是在python中做的方法。我专门问如何在C ++中做到这一点。
事实证明很简单：
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);
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句