当前位置:  开发笔记 > 编程语言 > 正文

C++ Tensorflow,如何使用多线程进行session-> Run(),或者花费更少的时间

如何解决《C++Tensorflow,如何使用多线程进行session->Run(),或者花费更少的时间》经验,为你挑选了0个好方法。

我在CPU上运行以下所有内容.我运行示例ensorflow/examples/label_image花费7~8秒.据我所知,python花费大约0.5秒来处理相同的样本,这是因为"TensorFlow会话对象是多线程的,因此多个线程可以轻松地使用相同的会话并且并行运行ops." 但是,如何使用C++在Session上设置多线程.

我尝试了...我在"tensorflow/tensorflow/core/common_runtime/direct_session.cc"中硬编码第81行:"const int32 num_threads = 16;" 但是,它不起作用.

我该如何设置一些配置或我应该做些什么?

推荐阅读
Gbom2402851125
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有