tf2.0 GPU被无限制占用

tensorflow-gpu在训练程序的时候会自动将GPU占满，解决方法有两种：

 

方法一：在程序开头加入

# GPU内存占用设置：方法1
physical_gpus = tf.config.list_physical_devices("GPU")
tf.config.experimental.set_virtual_device_configuration(
physical_gpus[0],
[tf.config.experimental.VirtualDeviceConfiguration(memory_limit=12000)]
)
logical_gpus = tf.config.list_logical_devices("GPU")

方法二：在程序开头加入

# GPU内存占用设置：方法2
physical_gpus = tf.config.list_physical_devices("GPU")
tf.config.experimental.set_memory_growth(physical_gpus[0],True)
logical_gpus = tf.config.list_logical_devices("GPU")

推荐使用第二种。