1つのコアでTensorflowを実行するにはどうすればよいですか？

Question

私はクラスタ上でTensorflowを使用しています。利用可能な場合でも、Tensorflowが単一のコアでのみ実行されるようにしたいと考えています。

これが可能かどうか誰かが知っていますか？

Answer 1

device_countのConfigProtoをセッションの作成時にconfig引数として渡すことで、TensorFlowが使用する特定の種類のデバイスの数を制限することができます。

config = tf.ConfigProto(device_count={'CPU': 1}) 
sess = tf.Session(config=config) 
with sess.as_default(): 
    print(tf.constant(42).eval()) 

Answer 2

1つのCPUスレッドでTensorflowを実行するには、私が使用：

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

device_countは、CPUの数が使用されている制限はなく、次のようにたとえば、あなたは、CPUデバイスの数を制限することができますコアまたはスレッドの数

tensorflow/tensorflow/core/protobuf/config.proto氏は述べています：Linuxの

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; 

---

を使用すると、例えば、あなたが持っているどのように多くのCPU /コア/スレッドを参照してsudo dmidecode -t 4 | egrep -i "Designation|Intel|core|thread"を実行することができますTensorflow 0.12.1及び1.0でテスト

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 

：以下は、それらの各々は、* 2 = 32 * 8 2スレッドの合計を与える2つのスレッドを有し、それらの各々は、8つのコアを有し、2つのCPUを有しています。 0とUbuntu 14.04.5 LTS x64とUbuntu 16.04 LTS x64。