How can I run Tensorflow on one single core?

by

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 limits the number of CPUs being used, not the number of cores or threads.

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

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;

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

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

More Related Contents:

  1. TensorFlow and person recognition in video stream
  2. How to understand static shape and dynamic shape in TensorFlow?
  3. How to Properly Combine TensorFlow’s Dataset API and Keras?
  4. How Can I Define Only the Gradient for a Tensorflow Subgraph?
  5. Get the value of some weights in a model trained by TensorFlow
  6. How do I disable TensorFlow’s eager execution?
  7. Confused by the behavior of `tf.cond`
  8. How to create dataset in the same format as the FSNS dataset?
  9. Split a dataset created by Tensorflow dataset API in to Train and Test?
  10. TensorFlow: slow performance when getting gradients at inputs
  11. TensorFlow, why there are 3 files after saving the model?
  12. Tensorflow doesn’t seem to see my gpu
  13. tensorflow on GPU: no known devices, despite cuda’s deviceQuery returning a “PASS” result
  14. Initializing tensorflow Variable with an array larger than 2GB
  15. TensorFlow: Dst tensor is not initialized
  16. Using Gensim Fasttext model with LSTM nn in keras
  17. What is the proper way to install TensorFlow on Apple M1 in 2022
  18. Simple way to visualize a TensorFlow graph in Jupyter?
  19. How do I convert a directory of jpeg images to TFRecords file in tensorflow?
  20. Tensorflow get all variables in scope
  21. Could not install packages due to an EnvironmentError: [WinError 5] Access is denied:
  22. Keras split train test set when using ImageDataGenerator
  23. CUDA_ERROR_OUT_OF_MEMORY in tensorflow
  24. Implementing contrastive loss and triplet loss in Tensorflow
  25. Confusion about keras Model: __call__ vs. call vs. predict methods
  26. What does batch, repeat, and shuffle do with TensorFlow Dataset?
  27. Checkpointing keras model: TypeError: can’t pickle _thread.lock objects
  28. How to replace (or insert) intermediate layer in Keras model?
  29. Choosing number of Steps per Epoch
  30. Parallelism isn’t reducing the time in dataset map

Leave a Comment