英文:
The kernel dies with jax.random.PGRNKey
问题
内核死机，出现了jax.random.PGRNKey的问题。
Python版本为3.10.10。
Jax版本为0.4.9，jaxlib版本为0.4.9。
在M2 MacBook上使用Jupyter Lab运行。
升级了软件包版本，但问题仍然存在。
对于它为何不断崩溃，我一无所知。
英文:
The kernel dies with jax.random.PGRNKey.
python version is 3.10.10.
jax version is 0.4.9. jaxlib version is 0.4.9.
It was run on M2 MacBook using Jupyter lab.
What should I do to make it work?
I upgraded the version of packages, and it didn't work.
I have no idea on why it keep crashing.
答案1
得分: 0
看起来 jaxlib 0.4.9 版本在 Mac ARM 上存在问题（参见JAX问题＃15951）。暂时建议安装 jax/jaxlib 版本0.4.8 来解决这个问题。
更新：jax和jaxlib 0.4.10已发布，应该修复了这个问题。
英文:
It looks like the jaxlib 0.4.9 release is broken on Mac ARM (see JAX Issue #15951). For now I'd recommend installing jax/jaxlib version 0.4.8 to fix the issue.
Update: jax & jaxlib 0.4.10 have been released, and should fix this issue.
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论