Skip to content

Commit 943bace

Browse files
authored
fix check for 12.1 (#1383)
1 parent 0b0d091 commit 943bace

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

conda/build_pytorch.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ else
265265
. ./switch_cuda_version.sh "$desired_cuda"
266266
# TODO, simplify after anaconda fixes their cudatoolkit versioning inconsistency.
267267
# see: https://github.com/conda-forge/conda-forge.github.io/issues/687#issuecomment-460086164
268-
if [[ "desired_cuda" == "12.1" ]]; then
268+
if [[ "$desired_cuda" == "12.1" ]]; then
269269
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=12.1,<12.2 # [not osx]"
270270
export MAGMA_PACKAGE=" - magma-cuda121 # [not osx and not win]"
271271
elif [[ "$desired_cuda" == "11.8" ]]; then

0 commit comments

Comments
 (0)