![从零开始学TensorFlow2.0](https://wfqqreader-1252317822.image.myqcloud.com/cover/188/33692188/b_33692188.jpg)
上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
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=1739381269-4rAPAJo5Dc7nTUHGhvQRGe0HhxRojVEO-0-4cbd98898175eef5b1090e03bb2fe4b4)
(2)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_2.jpg?sign=1739381269-tg7KJayvXaHJ5qny5mbDknyEWbB8wamD-0-7fbd2f9299843889b18a37bdda00dbda)
说明:Tensor.device提供托管张量内容的设备的完全限定字符串名称。该名称编码了许多详细信息,是分布式执行TensorFlow程序所必需的。
(3)在TensorFlow中,Placement指如何分配设备以执行各操作。如果没有明确指定,TensorFlow会自动决定执行操作的设备,并在需要时将张量复制到该设备。也可以使用tf.device上下文管理器将TensorFlow操作显式分配到特定设备上,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_3.jpg?sign=1739381269-15Ez1ukfQBOcyGMmkZ18B9Dpb0s4fHn1-0-2d8218d4a94676c4b347aa6b96be4305)
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_1.jpg?sign=1739381269-aEvexqTm0ONqFjk2cXMZX8hDpT8exIEp-0-03622086c602af33d61de6e210eccdc3)
(4)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_2.jpg?sign=1739381269-C3pMoaYESgcOCQdfQzfLCTlwTsZjxBcy-0-2fc67b2677e7cf5563c712e5654bb28c)