don't detect arch list when it has already been set & fix build-system requirments (#1802)

* fix cuda < 12.8 doesn't support 12.0

* build system doesn't need requirements

Author: CSY-ModelCloud

pyproject.toml

@@ -1,22 +1,3 @@
 [build-system]
-requires = ["setuptools >= 64",
-    # same with requirements.txt
-    "accelerate>=1.10.1" ,
-    "numpy>=2.2.6" ,
-    "torch>=2.8.0" ,
-    "safetensors>=0.6.2" ,
-    "transformers>=4.56.0" ,
-    "threadpoolctl>=3.6.0" ,
-    "packaging>=24.2" ,
-    "device-smi==0.4.1" ,
-    "protobuf>=6.32.0" ,
-    "pillow>=11.3.0" ,
-    "hf_transfer>=0.1.9" ,
-    "huggingface_hub>=0.34.4" ,
-    "random_word==1.0.13" ,
-    "tokenicer>=0.0.5" ,
-    "logbar==0.0.4" ,
-    "wheel>=0.45.1" ,
-    "maturin>=1.9.3" ,
-]
+requires = ["setuptools >= 64"]
 build-backend = "setuptools.build_meta:__legacy__"

setup.py

@@ -222,8 +222,9 @@ def get_version_tag() -> str:
 
 TORCH_VERSION = _read_env("TORCH_VERSION")
 RELEASE_MODE = _read_env("RELEASE_MODE")
-CUDA_VERSION = _read_env("TORCH_VERSION")
+CUDA_VERSION = _read_env("CUDA_VERSION")
 ROCM_VERSION = _read_env("ROCM_VERSION")
+TORCH_CUDA_ARCH_LIST = _read_env("TORCH_CUDA_ARCH_LIST")
 
 
 # respect user env then detect
@@ -257,7 +258,7 @@ def get_version_tag() -> str:
 # Handle CUDA_ARCH_LIST (public) and set TORCH_CUDA_ARCH_LIST for build toolchains
 CUDA_ARCH_LIST = _detect_cuda_arch_list() if (BUILD_CUDA_EXT == "1" and not ROCM_VERSION) else None
 
-if CUDA_ARCH_LIST:
+if not TORCH_CUDA_ARCH_LIST and CUDA_ARCH_LIST:
     archs = _parse_arch_list(CUDA_ARCH_LIST)
     kept = []
     for arch in archs: