TensorFlow(Mac OS X):无法确定CPU核心数:

Question

必须有一个简单的Mac OS X设置,以摆脱以下警告... .bash_profile中的东西？

>>> import tensorflow as tf
>>> sess = tf.Session()
can't determine number of CPU cores: assuming 4
I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 4

Answer 1

要为相关配置选项提供显式值,您可以执行以下操作:

NUM_CORES = ...  # Choose how many cores to use.
sess = tf.Session(
    config=tf.ConfigProto(inter_op_parallelism_threads=NUM_CORES,
                   intra_op_parallelism_threads=NUM_CORES))

此问题出现在TensorFlow for Mac OS X的初始二进制版本中,但应在此提交中修复:https://github.com/tensorflow/tensorflow/commit/430a054d6134f00e5188906bc4080fb7c5035ad5

该修复程序将包含在下一个二进制版本中.在此期间,您可以按照以下说明尝试从源代码构建:http://tensorflow.org/get_started/os_setup.md#installing_from_sources