Kernel dies
内核随jax.random.PGRNKey一起死亡。Python版本是3.10.10。jax版本是0.4.9。jaxlib版本是0.4.9。在M2 MacBook上使用Jupyter实验室运行。我该怎么做才能让它工作？
我升级了软件包的版本，它不工作。我不知道为什么它一直在崩溃。
Kernel dies
内核随jax.random.PGRNKey一起死亡。Python版本是3.10.10。jax版本是0.4.9。jaxlib版本是0.4.9。在M2 MacBook上使用Jupyter实验室运行。我该怎么做才能让它工作？
我升级了软件包的版本，它不工作。我不知道为什么它一直在崩溃。
1条答案
按热度按时间jyztefdp1#
看起来jaxlib 0.4.9版本在Mac ARM上被破坏了（参见JAX Issue #15951）。现在我建议安装jax/jaxlib版本0.4.8来解决这个问题。