tensorflow - 如何并行运行张量流模型？我需要重新编程模型吗？

Question

张量流中是否有一种简单的方法可以使任何模型并行运行而无需重新编程模型？就像在 PyTorch 中使用DataParallel方法一样

我是 tensorflow 的新手，因此非常感谢任何参考或代码示例。

标签： tensorflow

Answer 1

也许您可以修改 tensorflow 的配置原型并使用：

config = tf.ConfigProto(device_count={'GPU': GPU},
                        inter_op_parallelism_threads=NBR_THREADS,
                        intra_op_parallelism_threads=NBR_THREADS)

with tf.Session(config=config) as sess:
   <run_your_code>