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