We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6527d83 commit 1fc7cacCopy full SHA for 1fc7cac
.github/scripts/install_fbgemm.sh
@@ -12,7 +12,9 @@ echo "CHANNEL"
12
echo "$CHANNEL"
13
14
if [ "$CHANNEL" = "nightly" ]; then
15
+ ${CONDA_RUN} pip install numpy
16
${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/"$CU_VERSION"
17
elif [ "$CHANNEL" = "test" ]; then
18
19
${CONDA_RUN} pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/test/"$CU_VERSION"
20
fi
0 commit comments