如何在Android上运行tensorflow示例时设置线程号？

Question

正如我们所知，tensorflow的android演示是多线程的。那么我怎样才能把它改成单线程模式呢？

Answer 1

现在没有办法从Java那里做到这一点。在local_device.cc中自动选取符合设备上检测到的内核数量的内部操作并行性值。如果你只是想测试，你可以覆盖那里的价值。

手机上的并行操作当前总是为1 - 一次只能运行一个操作。

更一般地在本机代码中创建Session对象时，可以设置intra和inter-parallism。例如：

tensorflow::SessionOptions options; 
tensorflow::ConfigProto& config = options.config; 
config.set_intra_op_parallelism_threads(num_threads); 
Session session = tensorflow::NewSession(options); 

这将覆盖自动选取的值。一个例子见benchmark_model.cc。

如果你想从Java设置它，你将不得不添加一个参数到TensorFlowInferenceInterface.initializeTensorFlow()。

不过，我们很快就会切换到Java API，因此访问SessionOptions来设置值会更容易。