![从零开始学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=1739296049-Y3VVuTO2DwkyEEtYC9YWhbwIvWqDSNGw-0-8d469cc4060a85257c70c8fc1a3f80fc)
(2)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_2.jpg?sign=1739296049-xgtJdT6ZbHI7VrWIQv3RD25VBeOyf1Gz-0-01ef1ce31d94c7937a0a4bf0e684b713)
说明:Tensor.device提供托管张量内容的设备的完全限定字符串名称。该名称编码了许多详细信息,是分布式执行TensorFlow程序所必需的。
(3)在TensorFlow中,Placement指如何分配设备以执行各操作。如果没有明确指定,TensorFlow会自动决定执行操作的设备,并在需要时将张量复制到该设备。也可以使用tf.device上下文管理器将TensorFlow操作显式分配到特定设备上,代码如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_29_3.jpg?sign=1739296049-SIJsW1gs5xNNvvDFN7HTTcGZSw08ph8S-0-ec101acb349c8e8db10c0d5216dc5ee8)
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_1.jpg?sign=1739296049-XofvK6vJrxHIm1xYghdT2ODpisuQ6qNB-0-782203ca11ee2cb51ae22c2e1795b257)
(4)代码的运行结果如下。
![](https://epubservercos.yuewen.com/938466/18002370308013906/epubprivate/OEBPS/Images/39376_30_2.jpg?sign=1739296049-vwUPjKPIRV7roFAgAxIYBL0RZ0F0qVNv-0-df64d30931a7b7c80392f63ffbc08951)