Commit 8cdb5282 authored by rusty1s's avatar rusty1s
Browse files

fix conda build [ci skip]

parent 587d1d02
...@@ -39,4 +39,8 @@ echo "PyTorch $TORCH_VERSION+$CUDA_VERSION" ...@@ -39,4 +39,8 @@ echo "PyTorch $TORCH_VERSION+$CUDA_VERSION"
echo "- $CONDA_PYTORCH_CONSTRAINT" echo "- $CONDA_PYTORCH_CONSTRAINT"
echo "- $CONDA_CUDATOOLKIT_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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment