python - 分配 gpu 分数急切执行

Question

基本上，我在 eager 模式下运行强化学习模型，我需要限制每个进程将从 gpu 申请的内存量。在图形 API 中，这可以通过修改 tf.ConfigProto() 对象并使用所述配置对象创建 session 来实现。

但是，在eager api中，是没有session的。那么我的疑问是，在这种情况下我该如何管理 gpu 内存？

Answer 1

tf.enable_eager_execution()接受一个 config 参数，其值将是相同的 ConfigProto 消息。

因此，您应该能够使用它为每个进程设置相同的选项。

希望对您有所帮助。