From 14b3ea9afce8c0ec0129eb479a77afc568dfa2a0 Mon Sep 17 00:00:00 2001 From: atalman Date: Fri, 7 Jul 2023 09:48:49 -0700 Subject: [PATCH] Make sure CUDA_VERSION is set for conda-build --- conda/build_pytorch.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh index 1b3e13007..9aff26bcb 100755 --- a/conda/build_pytorch.sh +++ b/conda/build_pytorch.sh @@ -281,6 +281,9 @@ else export CONDA_TRITON_CONSTRAINT=" - torchtriton==2.1.0+${TRITON_SHORTHASH}" fi + # This to make sure conda-build is receiving correct CUDA_VERSION + export CUDA_VERSION="$desired_cuda" + build_string_suffix="cuda${CUDA_VERSION}_cudnn${CUDNN_VERSION}_${build_string_suffix}" fi