Update

[ghstack-poisoned]
pytorch · Oct 22, 2024 · d0287be · d0287be
1 parent 9be6bdf
commit d0287be
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/.github/unittest/linux_libs/scripts_rlhf/install.sh b/.github/unittest/linux_libs/scripts_rlhf/install.sh
@@ -31,15 +31,15 @@ git submodule sync && git submodule update --init --recursive
 printf "Installing PyTorch with cu121"
 if [[ "$TORCH_VERSION" == "nightly" ]]; then
   if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install --pre torch numpy==1.26.3 --index-url https://download.pytorch.org/whl/nightly/cpu -U
+      pip3 install --pre torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cpu -U
   else
-      pip3 install --pre torch numpy==1.26.3 --index-url https://download.pytorch.org/whl/nightly/cu121 -U
+      pip3 install --pre torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/nightly/cu121 -U
   fi
 elif [[ "$TORCH_VERSION" == "stable" ]]; then
     if [ "${CU_VERSION:-}" == cpu ] ; then
-      pip3 install torch numpy==1.26.3 --index-url https://download.pytorch.org/whl/cpu
+      pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cpu
   else
-      pip3 install torch numpy==1.26.3 --index-url https://download.pytorch.org/whl/cu121
+      pip3 install torch "numpy<2.0.0" --index-url https://download.pytorch.org/whl/cu121
   fi
 else
   printf "Failed to install pytorch"