jax.profiler.start_server

jax.profiler.start_server#

jax.profiler.start_server(port)[源代码][源代码]#

在端口 port 启动分析器服务器。

TensorBoard 2.2 或更高版本中使用“TensorFlow 分析器”功能,您可以连接到分析器服务器并采样显示 CPU、GPU 和/或 TPU 设备活动的执行轨迹。

参数:

port (int)

返回类型:

xla_client.profiler.ProfilerServer