可以在Tensorflow中默认设置'allow_growth=True'吗

我知道，对于Tensorflow，如果我们没有将

allow_growth

设置为

True

，GPU内存将完全映射到程序，以避免默认情况下出现内存碎片。但是，我正在构建一个多用户系统，它将支持多用户同时使用同一张卡。当然，我不希望GPU内存被完全分配，因此默认情况下，

allow_growth

设置为

True

将很容易解决问题

---

但是如何在Tensorflow外部默认设置

allow\u growth=True

？

此处记录了

allow\u growth

选项：

目前，配置

allow_growth

的方式是通过每个Python脚本中的Tensorflow会话选项

以前曾提出过通过环境变量进行配置的想法：

一般来说，由于Tensorflow是一个库，我们试图通过编程方式（如会话选项）保持配置。但是，如果您想在模型中默认关闭此功能，可以编写一个帮助器库，根据您的喜好进行设置

至于为什么Tensorflow默认值不是

允许增长

；我不确定。我怀疑这与碎片问题有关。您可以打开一个github问题，请求将其设置为默认值

---

希望有帮助

您可以尝试将环境变量

TF\u FORCE\u GPU\u ALLOW\u GROWTH

设置为值

true

。请注意，我在Tensorflow的代码中发现了这一点——它不是一个文档化的特性