- 从零开始学TensorFlow2.0
- 赵铭 欧铁军编著
- 241字
- 2021-03-29 04:53:02
3.2 GPU加速
使用GPU可以加速许多TensorFlow操作。如果没有任何注释,TensorFlow会自动决定是使用GPU还是CPU进行操作。
(1)张量可以在CPU和GPU内存之间进行复制,其代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_1.jpg?sign=1734493928-BvpJWbTRFs7Jvmuc9nr7LKMTI0E4nHKi-0-bc809ade5ab3e40e90dc93945a012c2f)
(2)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_2.jpg?sign=1734493928-RqafejtRN0WSBD4DDpd3nafpjT4cs89u-0-9ff902eefe55bf5ee909f89574b9c060)
说明:Tensor.device提供托管张量内容的设备的完全限定字符串名称。该名称编码了许多详细信息,是分布式执行TensorFlow程序所必需的。
(3)在TensorFlow中,Placement指如何分配设备以执行各操作。如果没有明确指定,TensorFlow会自动决定执行操作的设备,并在需要时将张量复制到该设备。也可以使用tf.device上下文管理器将TensorFlow操作显式分配到特定设备上,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_3.jpg?sign=1734493928-TKWVvZJUF47Izdlj93rTpBmUUI1z675N-0-511c524ebc7ae05aa7530329ed2ebec3)
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_1.jpg?sign=1734493928-32cZSC3B3m9YUx2zGt4yTuPovk3eNI4L-0-9ed2fbd5db4e135021b83791ba794d5d)
(4)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_2.jpg?sign=1734493928-7VVkwAEswhQANvNEIegDkcG4OXhzggZ6-0-c9dbefb9e25a817cc37f15e376ac4a90)