TensorFlow GPU优化与分布式训练指南.pdfVIP

GPU

设置

GPU

设置

◆默认使用全部

GPU

并占满所有内存

◆如何避免浪费内存和计算？

◆内存自动扩容

◆虚拟设备机制

◆多

GPU

使用

◆虚拟

GPU

与实际

GPU

◆手工设置

分布式机制

API

列表

◆tf.debugging.set_log_device_cement

某个变量分配在哪个设备上

__

◆tf.config.experimental.setvisibledevices

设置当前进程可见的设备

_

◆tf.config.experimental.ist

logicaldevices

获取逻辑设备

_

◆tf.config.experimental.list_physicaldevices

获取物理设备列表

__

◆tf.config.experimental.setmemorygrowth

按需分配显存

◆tf.config.experimental.VirtualDeviceConfigu