Install numpy 2.0 before building torchvision

pytorch · Jun 3, 2024 · f691287 · f691287
1 parent 569dad7
commit f691287
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/scripts/setup-env.sh b/.github/scripts/setup-env.sh
@@ -83,6 +83,10 @@ if [[ $GPU_ARCH_TYPE == 'cuda' ]]; then
 fi
 echo '::endgroup::'
 
+echo '::group::Install np2 before building tv'
+pip install "numpy==2.0.0rc2"
+echo '::endgroup::'
+
 echo '::group::Install third party dependencies prior to TorchVision install'
 # Installing with `easy_install`, e.g. `python setup.py install` or `python setup.py develop`, has some quirks when
 # when pulling in third-party dependencies. For example: