From bd1fffd9ffe755f8a8adb952099113c78929ff0f Mon Sep 17 00:00:00 2001
From: pbialecki <piotr.bialecki@hotmail.de>
Date: Thu, 6 Apr 2023 03:58:17 -0700
Subject: [PATCH] fix check for 12.1
---
conda/build_pytorch.sh | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
index b9bdcf89b..3dc150c7b 100755
--- a/conda/build_pytorch.sh
+++ b/conda/build_pytorch.sh
@@ -265,7 +265,7 @@ 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" == "12.1" ]]; then
+ if [[ "$desired_cuda" == "12.1" ]]; then
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=12.1,<12.2 # [not osx]"
export MAGMA_PACKAGE=" - magma-cuda121 # [not osx and not win]"
elif [[ "$desired_cuda" == "11.8" ]]; then