- 从零开始学TensorFlow2.0
- 赵铭 欧铁军编著
- 241字
- 2025-02-18 08:53:28
3.2 GPU加速
使用GPU可以加速许多TensorFlow操作。如果没有任何注释,TensorFlow会自动决定是使用GPU还是CPU进行操作。
(1)张量可以在CPU和GPU内存之间进行复制,其代码如下。

(2)代码的运行结果如下。

说明:Tensor.device提供托管张量内容的设备的完全限定字符串名称。该名称编码了许多详细信息,是分布式执行TensorFlow程序所必需的。
(3)在TensorFlow中,Placement指如何分配设备以执行各操作。如果没有明确指定,TensorFlow会自动决定执行操作的设备,并在需要时将张量复制到该设备。也可以使用tf.device上下文管理器将TensorFlow操作显式分配到特定设备上,代码如下。


(4)代码的运行结果如下。
