How to stop TensorFlow from multi-threading

前端 未结 1 634
予麋鹿
予麋鹿 2020-12-06 08:41

I am writing code for NIST FRVT. NIST wants the program to run at max 2 threads(Only CPU, No GPU). I am using TensorFlow in my code but it always spawns much more than 2 thr

相关标签:
1条回答
  • 2020-12-06 08:54

    With reference to the following sources, it looks like there is no possibility to run TensorFlow on 1 or 2 threads.

    • https://github.com/tensorflow/tensorflow/issues/33627
    • https://github.com/tensorflow/tensorflow/issues/42510
    • https://github.com/theislab/batchglm/issues/27
    • Using the script variable OMP_NUM_THREADS in the program source files
    • Change number of threads for Tensorflow inference with C API
    0 讨论(0)
提交回复
热议问题