如何查找有关运行程序的Cloud TPU设备的更多信息?

时间:2018-11-13 18:40:48

标签: tensorflow google-cloud-tpu

我们是使用Google Colab还是直接访问Cloud TPU,下面的程序仅提供有关基础TPU的有限信息:

function getToken_(){

  var securityModel = {"username":"serviceusername","password":"servicepassword"};

  var options = {
    'method' : 'POST',
    'contentType': 'application/json',
    'payload' : JSON.stringify(securityModel)
  };

  var tokenResponse = UrlFetchApp.fetch('http://myapiurlhere/token', options);

  return tokenResponse;
}

是否有任何以编程方式或通过bash命令显示的文档以显示更多信息,请参见此要点,例如https://gist.github.com/neomatrix369/256913dcf77cdbb5855dd2d7f5d81b84

1 个答案:

答案 0 :(得分:1)

Cloud TPU系统架构与GPU略有不同,因此此级别的信息不可用。

由于客户端与远程TensorFlow server进行对话并使用XLA,因此无需根据TPU上的可用功能更改客户端代码,因此远程服务器将根据以下内容编译机器指令: TPU的功能。

然而,Cloud TPU Profiler确实提供了TPU的较低级别视图,用于性能优化。您可以看到跟踪级别的视图,了解哪些操作正在消耗内存和计算时间。