The environment problem

[BotFu-coder]

When I was running the model inference, I ran into an environmental problem.
I use this code to install requirements:

pip install rlds tf_agents dm-reverb[tensorflow] apache_beam tfp-nightly

When I was running the code, I get this:

Traceback (most recent call last):
  File "/share/nfs/KGJ/RTX/open_x_embodiment-main/colabs/running.py", line 6, in <module>
    from tf_agents.policies import py_tf_eager_policy
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tf_agents/__init__.py", line 74, in <module>
    from tf_agents import agents
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tf_agents/agents/__init__.py", line 17, in <module>
    from tf_agents.agents import behavioral_cloning
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tf_agents/agents/behavioral_cloning/__init__.py", line 17, in <module>
    from tf_agents.agents.behavioral_cloning import behavioral_cloning_agent
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tf_agents/agents/behavioral_cloning/behavioral_cloning_agent.py", line 37, in <module>
    import tensorflow_probability as tfp
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tensorflow_probability/__init__.py", line 22, in <module>
    from tensorflow_probability.python import *  # pylint: disable=wildcard-import
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tensorflow_probability/python/__init__.py", line 152, in <module>
    dir(globals()[pkg_name])  # Forces loading the package from its lazy loader.
    ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tensorflow_probability/python/internal/lazy_loader.py", line 60, in __dir__
    module = self._load()
             ^^^^^^^^^^^^
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tensorflow_probability/python/internal/lazy_loader.py", line 40, in _load
    self._on_first_access()
  File "/home/linuxbrew/.linuxbrew/Cellar/[email protected]/3.11.3/lib/python3.11/site-packages/tensorflow_probability/python/__init__.py", line 59, in _validate_tf_environment
    raise ImportError(
ImportError: This version of TensorFlow Probability requires TensorFlow version >= 2.16; Detected an installation of version 2.15.0. Please upgrade TensorFlow to proceed.

So I update the tensorflow, but I get this:

ImportError: 
Attempted to load a reverb dynamic library, but it could not find the required
symbols inside of TensorFlow.  This commonly occurs when your version of
tensorflow and reverb are mismatched.  For example, if you are using the python
package 'tf-nightly', make sure you use the python package 'dm-reverb-nightly'
built on the same or the next night.  If you are using a release version package
'tensorflow', use a release package 'dm-reverb' built to be compatible with
that exact version.  If all else fails, file a github issue on deepmind/reverb.
Current installed version of tensorflow: 2.16.1.

I guess the reverb don't support tensorflow 2.16.1 , but I have no idea to fix that, hope someone can help, thx.