jax.profiler.start_server

jax.profiler.start_server(port)

在端口 port 上启动 profiler 服务器。

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

参数:

port (int)

返回类型:

xla_client.profiler.ProfilerServer