fix conda build [ci skip]

8cdb5282 · rusty1s · 587d1d02 · 8cdb5282
Commit 8cdb5282 authored Jun 30, 2022 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

conda/pytorch-sparse/build_conda.sh conda/pytorch-sparse/build_conda.sh +5 -1

No files found.
--- a/conda/pytorch-sparse/build_conda.sh
+++ b/conda/pytorch-sparse/build_conda.sh
@@ -39,4 +39,8 @@ echo "PyTorch $TORCH_VERSION+$CUDA_VERSION"
 echo "- $CONDA_PYTORCH_CONSTRAINT"
 echo "- $CONDA_CUDATOOLKIT_CONSTRAINT"

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