gpt4 book ai didi

tensorflow - 配置Tensorflow以使用所有CPU

转载 作者:行者123 更新时间:2023-12-03 23:25:49 25 4
gpt4 key购买 nike

读 :
https://www.tensorflow.org/versions/r0.10/resources/faq.html它指出:


TensorFlow是否使用所有可用的设备(GPU和CPU)
在我的机器上?

TensorFlow支持多个GPU和CPU。查看操作方法
有关如何在TensorFlow中使用GPU的文档
TensorFlow将操作分配给设备,以及CIFAR-10教程
对于使用多个GPU的示例模型。

请注意,TensorFlow仅使用具有计算能力的GPU设备
大于3.5。


这是否意味着Tensorflow可以自动使用给定机器上的所有CPU,或者它是否已明确配置?

最佳答案

CPU通过只是线程池的“设备”使用。如果您感觉需要更多,可以控制线程数:

sess = tf.Session(config=tf.ConfigProto(
intra_op_parallelism_threads=NUM_THREADS))

关于tensorflow - 配置Tensorflow以使用所有CPU,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39395198/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com