【问题标题】:TensorFlow (Mac OS X): can't determine number of CPU cores:TensorFlow (Mac OS X):无法确定 CPU 核心数:
【发布时间】:2016-02-10 14:26:02
【问题描述】:

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

【问题讨论】:

    标签: macos tensorflow


    【解决方案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

    【讨论】:

    • 没有按所写的那样工作。我需要使用tf.Session(config=tf.ConfigProto())
    猜你喜欢
    • 1970-01-01
    • 2013-04-04
    • 2016-02-12
    • 1970-01-01
    • 1970-01-01
    • 2012-03-13
    • 2015-03-02
    • 2010-12-15
    • 2016-05-09
    相关资源
    最近更新 更多