From 71f256bd52a375d8a8db0a8233296ba018568f3f Mon Sep 17 00:00:00 2001 From: Han Qi Date: Fri, 17 May 2024 22:23:53 +0000 Subject: [PATCH] update script --- install_everything.sh | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/install_everything.sh b/install_everything.sh index e81cd16f..71499877 100644 --- a/install_everything.sh +++ b/install_everything.sh @@ -16,12 +16,15 @@ pip show jax && pip uninstall -y jax pip show jaxlib && pip uninstall -y jaxlib pip show libtpu-nightly && pip uninstall -y libtpu-nightly +pip show tensorflow && pip uninstall -y tensorflow -pip install pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html +pip install jax[tpu]==0.4.28 -f https://storage.googleapis.com/jax-releases/libtpu_releases.html # torch cpu -pip install torch --index-url https://download.pytorch.org/whl/cpu +pip install torch==2.2.1+cpu --index-url https://download.pytorch.org/whl/cpu pip install tensorflow flatbuffers absl-py flax sentencepiece seqio google-cloud-storage pip install safetensors colorama coverage ray[default] humanize git submodule update --init --recursive +pip show google-jetstream && pip uninstall -y google-jetstream +pip show torch_xla2 && pip uninstall -y torch_xla2 pip install -e .