tensorflow 无法导入文件

quhf5bfb  于 2023-02-05  发布在  其他
关注(0)|答案(1)|浏览(157)

我尝试导入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 概率,结果都是一样的。

fnvucqvd

fnvucqvd1#

tensorflow_probability v0.14是在2021年9月发布的(history),当时JAX的最新版本是0.2.20(history),从那时起JAX已经发布了36个版本,所以出现一些不兼容的情况并不奇怪。
我尝试在谷歌Colab,发现以下组合工作:

import tensorflow
import tensorflow_probability
import jax
print(f"{jax.__version__=}")
print(f"{tensorflow.__version__=}")
print(f"{tensorflow_probability.__version__=}")

import tensorflow_probability.substrates.jax as tfp
print("loaded!")
jax.__version__='0.3.25'
tensorflow.__version__='2.9.2'
tensorflow_probability.__version__='0.17.0'
loaded!

另一种可能导致类似问题的情况是,如果你在笔记本电脑环境中工作,并安装已经导入的新版本的包。如果你在笔记本电脑中工作,请确保在安装或更新包后重新启动Python运行时。

相关问题