Skip to content

Commit 8cdb528

Browse files
committed
fix conda build [ci skip]
1 parent 587d1d0 commit 8cdb528

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

conda/pytorch-sparse/build_conda.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,4 +39,8 @@ echo "PyTorch $TORCH_VERSION+$CUDA_VERSION"
3939
echo "- $CONDA_PYTORCH_CONSTRAINT"
4040
echo "- $CONDA_CUDATOOLKIT_CONSTRAINT"
4141

42-
conda build . -c pytorch -c pyg -c default -c nvidia --output-folder "$HOME/conda-bld"
42+
if [ "${CUDA_VERSION}" = "cu116" ]; then
43+
conda build . -c pytorch -c default -c nvidia -c conda-forge --output-folder "$HOME/conda-bld"
44+
else
45+
conda build . -c pytorch -c default -c nvidia --output-folder "$HOME/conda-bld"
46+
fi

0 commit comments

Comments
 (0)