tf.config：GPU 的使用与分配（转载）

指定当前程序使用的 GPU 首先，通过 tf.config.list_physical_devices ，a56爆大奖在线娱乐们可以获得当前主机上某种特定运算设备类型（如 GPU 或 CPU ）的列表，例如，在一台具有 4 块 GPU 和一个 CPU 的工作站上运行以下代码： gpus = tf.config.lis