# To run
gcloud  compute tpus tpu-vm ssh --zone "us-central2-b" "node-3" --project "<GCP_PROJECT_NAME>"
# To scp datasets
gcloud compute tpus tpu-vm scp --recurse ~/arc/src/datasets/storage/ node-3:~/arc/src/datasets/

# To set up
sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt update
sudo apt upgrade -y
sudo apt-get remove python3.8 -y
sudo apt-get remove python3.9 -y
sudo apt install python3.11 python3.11-distutils -y
curl -sS https://bootstrap.pypa.io/get-pip.py | python3.11

git config --global credential.helper store
git clone ...
# token: ...

nano ~/.bashrc
alias python=python3.11
alias python3=python3.11
alias pip='python3.11 -m pip'
cd arc
export PYTHONPATH=${PYTHONPATH}:${PWD}
export HF_TOKEN=...
export WANDB_API_KEY=...
# save and close
source ~/.bashrc

pip install -U pip setuptools wheel
pip install -U -r requirements.txt "jax[tpu]" -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
