From 70dccaa84eb3dca4399e9204c043e83e906d4e01 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 15 Aug 2023 06:33:51 -0700 Subject: [PATCH 1/2] Add numpy install --- windows/internal/smoke_test.bat | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/windows/internal/smoke_test.bat b/windows/internal/smoke_test.bat index decb2cfb1..601005fff 100644 --- a/windows/internal/smoke_test.bat +++ b/windows/internal/smoke_test.bat @@ -87,6 +87,11 @@ set "PATH=%CONDA_HOME%;%CONDA_HOME%\scripts;%CONDA_HOME%\Library\bin;%PATH%" conda create -qyn testenv python=%DESIRED_PYTHON% if errorlevel 1 exit /b 1 +:: Install numpy see: https://github.com/pytorch/pytorch/issues/107228 +:: todo: Remove this install once the issue above is resolved +conda install -yq numpy +if errorlevel 1 exit /b 1 + call %CONDA_HOME%\condabin\activate.bat testenv if errorlevel 1 exit /b 1 From cfcdd9ea8492c33e2e248a582317c7eb8eb7e372 Mon Sep 17 00:00:00 2001 From: atalman Date: Tue, 15 Aug 2023 06:34:43 -0700 Subject: [PATCH 2/2] Add numpy install --- windows/internal/smoke_test.bat | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/windows/internal/smoke_test.bat b/windows/internal/smoke_test.bat index 601005fff..ad276b992 100644 --- a/windows/internal/smoke_test.bat +++ b/windows/internal/smoke_test.bat @@ -87,16 +87,13 @@ set "PATH=%CONDA_HOME%;%CONDA_HOME%\scripts;%CONDA_HOME%\Library\bin;%PATH%" conda create -qyn testenv python=%DESIRED_PYTHON% if errorlevel 1 exit /b 1 -:: Install numpy see: https://github.com/pytorch/pytorch/issues/107228 -:: todo: Remove this install once the issue above is resolved -conda install -yq numpy -if errorlevel 1 exit /b 1 - call %CONDA_HOME%\condabin\activate.bat testenv if errorlevel 1 exit /b 1 :: do conda install to make sure all the dependencies are installed -call conda install -yq pytorch %CONDA_EXTRA_ARGS% +:: Install numpy see: https://github.com/pytorch/pytorch/issues/107228 +:: todo: Remove numpy install once the issue above is resolved +call conda install -yq numpy pytorch %CONDA_EXTRA_ARGS% if ERRORLEVEL 1 exit /b 1 set /a CUDA_VER=%CUDA_VERSION%