Tensorboard未显示GPU设备

时间:2019-03-17 23:56:40

标签: python tensorflow tensorboard

我正在使用TensorFlow版本1.13。 我使用GPU训练了2种不同的模型:

  • NVIDIA V100:Adanet
  • NVIDIA K80:MNIST

在第一个Adanet模型中,在培训期间,我看到TF正在看到GPU:

01.03.2002

在TensorBoard中:在Graphs-> Device->下,我没有看到我的GPU。

enter image description here

注意:我正在使用GCP Cloud ML引擎分配GPU

“ XLA群集”,“计算时间”,“内存”显示为灰色。

我需要在TensorFlow代码中添加任何特定参数吗?

0 个答案:

没有答案