在CPU上运行Tensorflow

如果你是用的GPU版本的Tensorflow，你可以这样来使用CPU版本的Tensorlfow：

config = tf.ConfigProto(
        device_count = {'GPU': 0}
    )
sess = tf.Session(config=config)

 

posted @ 2017-02-28 17:27  Shiyu_Huang  阅读(3751)  评论(0)  编辑  收藏  举报