上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
3.2 GPU加速
使用GPU可以加速许多TensorFlow操作。如果没有任何注释,TensorFlow会自动决定是使用GPU还是CPU进行操作。
(1)张量可以在CPU和GPU内存之间进行复制,其代码如下。
(2)代码的运行结果如下。
说明:Tensor.device提供托管张量内容的设备的完全限定字符串名称。该名称编码了许多详细信息,是分布式执行TensorFlow程序所必需的。
(3)在TensorFlow中,Placement指如何分配设备以执行各操作。如果没有明确指定,TensorFlow会自动决定执行操作的设备,并在需要时将张量复制到该设备。也可以使用tf.device上下文管理器将TensorFlow操作显式分配到特定设备上,代码如下。
(4)代码的运行结果如下。