jax.profiler.start_server#
- jax.profiler.start_server(port)[source]#
在端口 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