I had this problem too. What helped me was



Ideas for beginner-friendlier TPU-VM clm training


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 Beginners
  
  
    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…