Skip to content

Commit 1fc7cac

Browse files
authored
Install numpy from pypi instead of pytorch.org
1 parent 6527d83 commit 1fc7cac

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/scripts/install_fbgemm.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@ echo "CHANNEL"
1212
echo "$CHANNEL"
1313

1414
if [ "$CHANNEL" = "nightly" ]; then
15+
${CONDA_RUN} pip install numpy
1516
${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/"$CU_VERSION"
1617
elif [ "$CHANNEL" = "test" ]; then
18+
${CONDA_RUN} pip install numpy
1719
${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/"$CU_VERSION"
1820
fi

0 commit comments

Comments
 (0)