AttributeError: module 'tensorflow.python.ops.linalg.linear_operator_util' has no attribute 'matmul_with_broadcast'

Question

我正在尝试为 TF-Agents 创建我自己的 PyEnvironment。但是，此错误不断出现：

AttributeError: module 'tensorflow.python.ops.linalg.linear_operator_util' has no attribute 'matmul_with_broadcast'

我发现这似乎是 tensorflow-probability 的问题，但我已经安装了推荐的版本 tensorflow-probability=0.7.0 https://github.com/tensorflow/agents/issues/91

我试过重新安装和更新

tensorflow-gpu=2.0.0-beta1 tf-agents-nightly tensorflow-probability=0.7.0

这是一个最小的代码示例：

from tf_agents.environments import py_environment


class myEnv(py_environment.PyEnvironment):
    def __init__(self):
        pass

    def _reset(self):
        pass

    def _step(self, action):
        pass

这是运行这个最小示例时的完整错误消息：

Traceback (most recent call last):   File ".\env_mwe.py", line 1, in <module>
    from tf_agents.environments import py_environment   File "C:\Python37\lib\site-packages\tf_agents\environments\__init__.py", line 18, in <module>
    from tf_agents.environments import batched_py_environment   File "C:\Python37\lib\site-packages\tf_agents\environments\batched_py_environment.py", line 33, in <module>
    from tf_agents.environments import py_environment   File "C:\Python37\lib\site-packages\tf_agents\environments\py_environment.py", line 29, in <module>
    from tf_agents.trajectories import time_step as ts   File "C:\Python37\lib\site-packages\tf_agents\trajectories\__init__.py", line 19, in <module>
    from tf_agents.trajectories import time_step   File "C:\Python37\lib\site-packages\tf_agents\trajectories\time_step.py", line 28, in <module>
    from tf_agents.specs import array_spec   File "C:\Python37\lib\site-packages\tf_agents\specs\__init__.py", line 20, in <module>
    from tf_agents.specs.distribution_spec import DistributionSpec   File "C:\Python37\lib\site-packages\tf_agents\specs\distribution_spec.py", line 22, in <module>
    import tensorflow_probability as tfp   File "E:\Users\tmp\AppData\Roaming\Python\Python37\site-packages\tensorflow_probability\__init__.py", line 78, in <module>
    from tensorflow_probability.python import *  # pylint: disable=wildcard-import   File "E:\Users\tmp\AppData\Roaming\Python\Python37\site-packages\tensorflow_probability\python\__init__.py", line 22, in <module>
    from tensorflow_probability.python import distributions   File "E:\Users\tmp\AppData\Roaming\Python\Python37\site-packages\tensorflow_probability\python\distributions\__init__.py", line 64, in <module>
    from tensorflow_probability.python.distributions.linear_gaussian_ssm import LinearGaussianStateSpaceModel   File "E:\Users\tmp\AppData\Roaming\Python\Python37\site-packages\tensorflow_probability\python\distributions\linear_gaussian_ssm.py", line 41, in <module>
    _matmul = linear_operator_util.matmul_with_broadcast AttributeError: module 'tensorflow.python.ops.linalg.linear_operator_util' has no attribute 'matmul_with_broadcast'

Answer 1

找到了！

为了将 tensorflow-gpu 与 tensorflow-probability 一起使用，需要安装 tfp-nightly

所以pip3 install --upgrade tfp-nightly成功了

Answer 2

要使用 2.0.0beta1 需要 tfp-nightly。

AttributeError: module 'tensorflow.python.ops.linalg.linear_operator_util' has no attribute 'matmul_with_broadcast'

AttributeError: module 'tensorflow.python.ops.linalg.linear_operator_util' has no attribute 'matmul_with_broadcast'

python

tensorflow2.0

tensorflow-probability

tensorflow-agents