代码之家 › 专栏 › 技术社区 › user3391229

tensorflow slim并行训练和评估回路;单个设备

tf-slim tensorflow-gpu tensorflow python-2.7

user3391229 · 技术社区 · 7 年前

例如,如果我有2个gpu,我可以在第一个gpu上运行一个以“slim.learning.train()”结尾的python脚本,在第二个gpu上运行另一个以“slim.evaluation.evaluation\u loop()”结尾的python脚本。

是否有一种方法可以为两个任务管理1个gpu的资源?tf。火车我想起了主管,但我真的不知道。

1 回复 | 直到 7 年前

Shruthi Sampath Kumar 6 年前

您可以使用以下代码对GPU使用情况进行分区。

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.3000) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) sess.run(tf.app.run())

推荐文章

MLNINJA · TensorFlow PR曲线插件:PR\u Curve\u streaming\u op

7 年前

Ahmed Khairy · AttributeError:模块的tensorflow。蟒蛇训练“training”没有属性“SummaryWriter”

7 年前

Schilcote · Tensorflow没有常数(但有其他的一切?)

7 年前

crucialoil · Windows 10上Keras的“ValueError:bad marshal data”

7 年前

devforfu · 在GPU上运行的TensorFlow Word2Vec模型

7 年前

DQ_happy · 为tf归一化的图像像素值。形象解码jpeg和tf。火车shuffle\u批次?

7 年前

Ujjwal · 使用tf。cond()为训练和验证提供图形

7 年前

j35t3r · 在Intellij Idea中运行tensorflow项目时找不到cuda8库

7 年前

Kong · TensorFlow后端不使用GPU的Keras

7 年前

user3391229 · tensorflow slim并行训练和评估回路;单个设备

7 年前