TensorFlow:如何验证它是否在 GPU 上运行

Question

我正在寻找一种简单的方法来验证我的 TF图实际上是在 GPU 上运行的。

附注。验证 cuDNN 也很好图书馆被使用。

Answer 1

有多种方法可以查看操作位置。

将 RunOptions 和 RunMetadata 添加到 session 调用中，并在 Tensorboard 中查看操作和计算的位置。请参阅此处的代码:https://www.tensorflow.org/get_started/graph_viz

在 session ConfigProto 中指定 log_device_placement 选项。这会记录到控制台操作被放置在哪个设备上。 https://www.tensorflow.org/api_docs/python/tf/ConfigProto

使用 nvidia-smi 在终端中查看 GPU 使用情况。