Install numpy from pypi instead of pytorch.org

pytorch · Jan 22, 2024 · 1fc7cac · 1fc7cac
1 parent 6527d83
commit 1fc7cac
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh
@@ -12,7 +12,9 @@ echo "CHANNEL"
 echo "$CHANNEL"
 
 if [ "$CHANNEL" = "nightly" ]; then
+    ${CONDA_RUN} pip install numpy
     ${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/"$CU_VERSION"
 elif [ "$CHANNEL" = "test" ]; then
+    ${CONDA_RUN} pip install numpy
     ${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/"$CU_VERSION"
 fi