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