Flax - core dump when starting training

[Norod78]

I had this problem too. What helped me was

@Norod78

pip install --upgrade clu

Also, if it’s relevant, I’ve wrote down this list of things I did which allowed me to train directly on the TPU VM:

Ideas for beginner-friendlier TPU-VM clm training

Follow this tutorial to set-up and connect to your TPU-VM Add your local bin path to the PATH environment variable, If you do not know your local user name, type: whoami #In my case, the user is ‘dadler’, so replace ‘dadler’ in the following block with your own user: nano ~/.bashrc #Add the following line at the bottom, replace dadler with your own user name export PATH="/home/dadler/.local/bin:$PATH" #Save (CTRL O) #Exit (Ctrl X) #Realod bashrc source ~/.bashrc Install and upgra…