error when run the training code

[Holmes-Alan]

I tried to install the code following the same procedure. However, I got the error when installing Jax, Jaxlib and Flax as follows.

raise RuntimeError(msg)
RuntimeError: jaxlib is version 0.1.57, but this version of jax requires version >= 0.3.7.

I checked that I installed Jax==0.3.9 and Jaxlib==0.1.57, which meets the requirement. I do not know what the problem is. I am looking forward to your reply.

[LolaDeng]

How did you resolve the issue?