Move numpy pin to 2.0.2 (#1976)

pytorch · Sep 3, 2024 · 543e421 · 543e421
1 parent 64eb2ff
commit 543e421
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/aarch64_linux/aarch64_ci_setup.sh b/aarch64_linux/aarch64_ci_setup.sh
@@ -28,7 +28,7 @@ conda activate "${CONDA_ENV_NAME}"
 if [[ "$DESIRED_PYTHON"  == "3.8" ]]; then
     pip install -q numpy==1.24.4
 else
-    pip install -q --pre numpy==2.0.0rc1
+    pip install -q --pre numpy==2.0.2
 fi
 conda install -y -c conda-forge pyyaml==6.0.1 patchelf==0.17.2 pygit2==1.13.2 ninja==1.11.1 scons==4.5.2
 

diff --git a/manywheel/build_common.sh b/manywheel/build_common.sh
@@ -125,11 +125,11 @@ case ${DESIRED_PYTHON} in
     retry pip install -q numpy==1.15
     ;;
   cp31*)
-    retry pip install -q --pre numpy==2.0.0rc1
+    retry pip install -q --pre numpy==2.0.2
     ;;
   # Should catch 3.9+
   *)
-    retry pip install -q --pre numpy==2.0.0rc1
+    retry pip install -q --pre numpy==2.0.2
     ;;
 esac
 
@@ -475,11 +475,11 @@ if [[ -z "$BUILD_PYTHONLESS" ]]; then
   fi
 
   pip uninstall -y "$TORCH_PACKAGE_NAME"
-  
+
   if [[ "$USE_SPLIT_BUILD" == "true" ]]; then
     pip install "$TORCH_NO_PYTHON_PACKAGE_NAME" --no-index -f /$WHEELHOUSE_DIR --no-dependencies -v
   fi
-  
+
   pip install "$TORCH_PACKAGE_NAME" --no-index -f /$WHEELHOUSE_DIR --no-dependencies -v
 
   # Print info on the libraries installed in this wheel

diff --git a/windows/internal/smoke_test.bat b/windows/internal/smoke_test.bat
@@ -54,10 +54,10 @@ if errorlevel 1 exit /b 1
 set "PATH=%CD%\Python%PYTHON_VERSION%\Scripts;%CD%\Python;%PATH%"
 
 
-if "%DESIRED_PYTHON%" == "3.12" pip install -q --pre numpy==2.0.0rc1 protobuf
-if "%DESIRED_PYTHON%" == "3.11" pip install -q --pre numpy==2.0.0rc1 protobuf
-if "%DESIRED_PYTHON%" == "3.10" pip install -q --pre numpy==2.0.0rc1 protobuf
-if "%DESIRED_PYTHON%" == "3.9" pip install -q --pre numpy==2.0.0rc1 protobuf
+if "%DESIRED_PYTHON%" == "3.12" pip install -q --pre numpy==2.0.2 protobuf
+if "%DESIRED_PYTHON%" == "3.11" pip install -q --pre numpy==2.0.2 protobuf
+if "%DESIRED_PYTHON%" == "3.10" pip install -q --pre numpy==2.0.2 protobuf
+if "%DESIRED_PYTHON%" == "3.9" pip install -q --pre numpy==2.0.2 protobuf
 if "%DESIRED_PYTHON%" == "3.8" pip install -q numpy protobuf
 
 if errorlevel 1 exit /b 1