tensorflow2.0——显存相关设置
# 设置相关底层配置 physical_devices = tf.config.experimental.list_physical_devices('GPU') assert len(physical_devices) > 0, "Not enough GPU hardware devices available" tf.config.experimental.set_memory_growth(physical_devices[0], True)
不设置会导致卷积相关代码无法运行
- 上一篇 »设置tensorflow的显存为动态使用
- 下一篇 »TensorFlow设置GPU占用量