Update libtpu nightly installation

AI-Hypercomputer · Oct 23, 2024 · 99e1bdc · 99e1bdc
1 parent 49822a3
commit 99e1bdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.sh b/setup.sh
@@ -180,7 +180,7 @@ elif [[ $MODE == "nightly" ]]; then
         else
             # Install libtpu-nightly
             echo "Installing libtpu-nightly"
-            pip3 install libtpu-nightly -f https://storage.googleapis.com/jax-releases/libtpu_releases.html -U --pre
+            pip3 install -U --pre libtpu -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
         fi
         echo "Installing nightly tensorboard plugin profile"
         pip3 install tbp-nightly --upgrade