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