From 1fc7cace40b425d8556e6d3108a06466adecc40b Mon Sep 17 00:00:00 2001
From: henrylhtsang <91030427+henrylhtsang@users.noreply.github.com>
Date: Mon, 22 Jan 2024 11:36:21 -0800
Subject: [PATCH] Install numpy from pypi instead of pytorch.org

---
 .github/scripts/install_fbgemm.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/scripts/install_fbgemm.sh b/.github/scripts/install_fbgemm.sh
index 08765d62f..cc4cfe9b6 100644
--- 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