diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh index 9772c1fb0..ad9cc6e52 100755 --- a/conda/build_pytorch.sh +++ b/conda/build_pytorch.sh @@ -267,11 +267,11 @@ else # TODO, simplify after anaconda fixes their cudatoolkit versioning inconsistency. # see: https://github.com/conda-forge/conda-forge.github.io/issues/687#issuecomment-460086164 if [[ "$desired_cuda" == "11.8" ]]; then - export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.9,<12.0 # [not osx]" - export MAGMA_PACKAGE=" - magma-cuda118 # [not osx and not win]" + export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.9,<12.0 # [not osx]" + export MAGMA_PACKAGE=" - magma-cuda118 # [not osx and not win]" elif [[ "$desired_cuda" == "11.7" ]]; then - export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.7,<11.8 # [not osx]" - export MAGMA_PACKAGE=" - magma-cuda117 # [not osx and not win]" + export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.7,<11.8 # [not osx]" + export MAGMA_PACKAGE=" - magma-cuda117 # [not osx and not win]" elif [[ "$desired_cuda" == "11.6" ]]; then export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.6,<11.7 # [not osx]" export MAGMA_PACKAGE=" - magma-cuda116 # [not osx and not win]"