如何在 tensorflow 2.5 中运行 tensorflow 分析

Posted 2023-02-16

【中文标题】如何在 tensorflow 2.5 中运行 tensorflow 分析

【英文标题】：How do I run tensorflow profiling in tensorflow 2.5

【发布时间】：2021-10-06 20:43:08

【问题描述】：

我在 Big Sur 11.5 上使用 Anaconda 运行 python 3.85

% conda list tensor
# packages in environment at /Users/davidlaxer/anaconda3/envs/ai:
#
# Name                    Version                   Build  Channel
tensor2tensor             1.15.7                   pypi_0    pypi
tensorboard               2.5.0                    pypi_0    pypi
tensorboard-data-server   0.6.1                    pypi_0    pypi
tensorboard-plugin-profile 2.4.0                    pypi_0    pypi
tensorboard-plugin-wit    1.8.0                    pypi_0    pypi
tensorflow                2.5.0                    pypi_0    pypi
tensorflow-datasets       4.3.0                    pypi_0    pypi
tensorflow-estimator      2.5.0                    pypi_0    pypi
tensorflow-hub            0.12.0                   pypi_0    pypi
tensorflow-metadata       1.0.0                    pypi_0    pypi
tensorflow-probability    0.13.0rc0                pypi_0    pypi
tensorflow-text           2.5.0                    pypi_0    pypi

这里是测试代码：

import tensorflow as tf
import numpy as np
from utils import *
from VDSH import *

tf.compat.v1.enable_v2_behavior()
from tensorflow.python.framework.ops import disable_eager_execution
disable_eager_execution()

#gpu_devices = tf.config.experimental.list_physical_devices('GPU')
#tf.config.experimental.set_memory_growth(gpu_devices[0], True)
options = tf.profiler.experimental.ProfilerOptions(host_tracer_level = 3,
                                                   python_tracer_level = 1,
                                                   device_tracer_level = 1)
tf.profiler.experimental.start('~/logdir', options=options)
...
tf.profiler.experimental.stop()

这是错误信息：

AttributeError: 模块 'tensorflow.compat.v1.profiler' 没有属性 'experimental'

【问题讨论】：

为什么启用v2_behaviour？在 Tensorflow 2.x 中，默认启用 Eager Execution。

为什么再次禁用急切执行？您可以在没有tf.compat.v1.enable_v2_behavior() from tensorflow.python.framework.ops import disable_eager_execution disable_eager_execution() 的情况下尝试上面的代码并告诉我们吗？

文件“/Users/davidlaxer/VariationalDeepSemanticHashing/test.py”，第 14 行，在 中 options = tf.profiler.experimental.ProfilerOptions(host_tracer_level = 3, AttributeError: module 'tensorflow.compat .v1.profiler' 没有属性 'experimental'

【参考方案1】：

要在 tensorboard 中显示 Profile 标签需要 Chrome。