From bd41ed80a2e99d5f8abb3b7e651256d0c4fbc689 Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Sun, 8 Jun 2025 06:30:50 +0900 Subject: [PATCH] [WIP] update linux workflow --- build_linux.sh | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/build_linux.sh b/build_linux.sh index 2949824..fe2567a 100755 --- a/build_linux.sh +++ b/build_linux.sh @@ -36,8 +36,10 @@ TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \ '2.7': [118, 126, 128], \ '2.8': [128], \ }; \ - target_cuda_versions = support_cuda_versions[env['MATRIX_TORCH_VERSION']]; \ - cuda_version = int(env['MATRIX_CUDA_VERSION']); \ + matrix_cuda_version = '$MATRIX_CUDA_VERSION'; \ + matrix_torch_version = '$MATRIX_TORCH_VERSION'; \ + target_cuda_versions = support_cuda_versions[matrix_torch_version]; \ + cuda_version = int(matrix_cuda_version); \ closest_version = min(target_cuda_versions, key=lambda x: abs(x - cuda_version)); \ print(closest_version) \ ")