Tensorflow中的GPU分配方法

Tensorflow中的GPU分配方法 默認狀況下 TensorFlow 會使用其所可以使用的全部 GPU，這樣，會出現浪費的狀況。html 列出當前設備上的GPU和CPU 首先，經過 tf.config.experimental.list_physical_devices，咱們能夠得到當前主機上某種特定運算設備類型（如 GPU 或 CPU ）的列表，例如，在一臺具備 4 塊 GPU 和一個 CP

>>阅读原文<<