From 51c181a9038a7a3000f9c06ad307ff98e0f5e160 Mon Sep 17 00:00:00 2001 From: pbialecki Date: Wed, 23 Nov 2022 22:23:27 -0800 Subject: [PATCH] add conda builds for CUDA 11.8 --- conda/build_pytorch.sh | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh index 7d130e8c0..9772c1fb0 100755 --- a/conda/build_pytorch.sh +++ b/conda/build_pytorch.sh @@ -266,7 +266,10 @@ else . ./switch_cuda_version.sh "$desired_cuda" # 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.7" ]]; then + 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]" + 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]" elif [[ "$desired_cuda" == "11.6" ]]; then