如何在单核上运行 Tensorflow? [英] How can I run Tensorflow on one single core?

问题描述

我在集群上使用 Tensorflow，我想告诉 Tensorflow 只在一个单核上运行(即使还有更多可用核).

I'm using Tensorflow on a cluster and I want to tell Tensorflow to run only on one single core (even though there are more available).

有人知道这是否可行吗?

Does someone know if this is possible?

推荐答案

要在单个 CPU 线程上运行 Tensorflow，我使用:

To run Tensorflow on one single CPU thread, I use:

session_conf = tf.ConfigProto(
      intra_op_parallelism_threads=1,
      inter_op_parallelism_threads=1)
sess = tf.Session(config=session_conf)

device_count 限制正在使用的 CPU 数量，而不是内核或线程数.

device_count limits the number of CPUs being used, not the number of cores or threads.

/tensorflow/core/protobuf/config.proto 说:

message ConfigProto {
  // Map from device type name (e.g., "CPU" or "GPU" ) to maximum
  // number of devices of that type to use.  If a particular device
  // type is not found in the map, the system picks an appropriate
  // number.
  map<string, int32> device_count = 1;

<小时>

在 Linux 上你可以运行 sudo dmidecode -t 4 |egrep -i "Designation|Intel|core|thread" 查看你有多少 CPU/内核/线程，例如下面有2个CPU，每个有8个内核，每个有2个线程，总共有2*8*2=32个线程:

---

On Linux you can run sudo dmidecode -t 4 | egrep -i "Designation|Intel|core|thread" to see how many CPUs/cores/threads you have, e.g. the following has 2 CPUs, each of them has 8 cores, each of them has 2 threads, which gives a total of 2*8*2=32 threads:

fra@s:~$ sudo dmidecode -t 4 | egrep -i "Designation|Intel|core|thread"
    Socket Designation: CPU1
    Manufacturer: Intel
            HTT (Multi-threading)
    Version: Intel(R) Xeon(R) CPU E5-2667 v4 @ 3.20GHz
    Core Count: 8
    Core Enabled: 8
    Thread Count: 16
            Multi-Core
            Hardware Thread
    Socket Designation: CPU2
    Manufacturer: Intel
            HTT (Multi-threading)
    Version: Intel(R) Xeon(R) CPU E5-2667 v4 @ 3.20GHz
    Core Count: 8
    Core Enabled: 8
    Thread Count: 16
            Multi-Core
            Hardware Thread

在 Ubuntu 14.04.5 LTS x64 和 Ubuntu 16.04 LTS x64 上使用 Tensorflow 0.12.1 和 1.0.0 进行测试.

Tested with Tensorflow 0.12.1 and 1.0.0 with Ubuntu 14.04.5 LTS x64 and Ubuntu 16.04 LTS x64.

这篇关于如何在单核上运行 Tensorflow?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！