我尝试导入tensorflow_probability. substrates. jax(特别是为了使用发行版)并得到如下所示的错误(看起来像是自导入)。我安装了tensorflow(2.8.2),tensorflow-probability(0.14.0)和jax(0.3.25)。
尝试
import tensorflow_probability.substrates.jax as tfp
我得到
ImportError: cannot import name 'bijectors' from partially initialized module
'tensorflow_probability.substrates.jax' (most likely due to a circular import)
(/path-to-anaconda3-env/lib/python3.10/site-packages/tensorflow_probability/substrates/jax/__init__.py)
我尝试了几个不同版本的tensorflow 概率,结果都是一样的。
1条答案
按热度按时间fnvucqvd1#
tensorflow_probability v0.14是在2021年9月发布的(history),当时JAX的最新版本是0.2.20(history),从那时起JAX已经发布了36个版本,所以出现一些不兼容的情况并不奇怪。
我尝试在谷歌Colab,发现以下组合工作:
另一种可能导致类似问题的情况是,如果你在笔记本电脑环境中工作,并安装已经导入的新版本的包。如果你在笔记本电脑中工作,请确保在安装或更新包后重新启动Python运行时。