3.2 GPU加速

使用GPU可以加速許多TensorFlow操作。如果沒有任何注釋，TensorFlow會(huì)自動(dòng)決定是使用GPU還是CPU進(jìn)行操作。

（1）張量可以在CPU和GPU內(nèi)存之間進(jìn)行復(fù)制，其代碼如下。

（2）代碼的運(yùn)行結(jié)果如下。

說明：Tensor.device提供托管張量?jī)?nèi)容的設(shè)備的完全限定字符串名稱。該名稱編碼了許多詳細(xì)信息，是分布式執(zhí)行TensorFlow程序所必需的。

（3）在TensorFlow中，Placement指如何分配設(shè)備以執(zhí)行各操作。如果沒有明確指定，TensorFlow會(huì)自動(dòng)決定執(zhí)行操作的設(shè)備，并在需要時(shí)將張量復(fù)制到該設(shè)備。也可以使用tf.device上下文管理器將TensorFlow操作顯式分配到特定設(shè)備上，代碼如下。

（4）代碼的運(yùn)行結(jié)果如下。