Remove cuda toolkit var settings

conda-forge · Dec 6, 2024 · 8c4fef0 · 8c4fef0
1 parent 4d27506
commit 8c4fef0
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/recipe/build.sh b/recipe/build.sh
@@ -11,21 +11,6 @@ if [[ ${cuda_compiler_version} != "None" ]]; then
     export TORCH_CUDA_ARCH_LIST="5.0;6.0;6.1;7.0;7.5;8.0;8.6;8.9;9.0+PTX"
     # $CUDA_HOME not set in CUDA 12.0. Using $PREFIX
     export CUDA_TOOLKIT_ROOT_DIR="${PREFIX}"
-    # PyTorch has multiple different bits of logic finding CUDA, override
-    # all of them.
-    export CUDAToolkit_BIN_DIR=${BUILD_PREFIX}/bin
-    export CUDAToolkit_ROOT_DIR=${PREFIX}
-    case ${target_platform} in
-        linux-64)
-            export CUDAToolkit_TARGET_DIR=${PREFIX}/targets/x86_64-linux
-            ;;
-        linux-aarch64)
-            export CUDAToolkit_TARGET_DIR=${PREFIX}/targets/sbsa-linux
-            ;;
-        *)
-            echo "unknown CUDA arch, edit build.sh"
-            exit 1
-    esac
     # CUDA_HOME must be set for the build to work in torchaudio
     export CUDA_HOME="${PREFIX}"
   else