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