jax.profiler.start_server#
- jax.profiler.start_server(port)[源代码]#
在端口 port 上启动性能分析器服务器。
使用 TensorBoard 2.2 或更新版本中的“TensorFlow 性能分析器”功能，您可以连接到性能分析器服务器并采样执行跟踪，以显示 CPU、GPU 和/或 TPU 设备活动。
- 参数：
port (int)
- 返回类型：
xla_client.profiler.ProfilerServer
在端口 port 上启动性能分析器服务器。
使用 TensorBoard 2.2 或更新版本中的“TensorFlow 性能分析器”功能，您可以连接到性能分析器服务器并采样执行跟踪，以显示 CPU、GPU 和/或 TPU 设备活动。
port (int)
xla_client.profiler.ProfilerServer