Tensorflow中的GPU分配方法

Tensorflow中的GPU分配方法 默认状况下 TensorFlow 会使用其所可以使用的全部 GPU，这样，会出现浪费的状况。html 列出当前设备上的GPU和CPU 首先，经过 tf.config.experimental.list_physical_devices，咱们能够得到当前主机上某种特定运算设备类型（如 GPU 或 CPU ）的列表，例如，在一台具备 4 块 GPU 和一个 CP

>>阅读原文<<