filter torch cuda arch < 6.0 (#955)

* filter arch < 6.0

* remove unused codes

Author: CSY-ModelCloud

setup.py

@@ -17,6 +17,10 @@
 
 TORCH_CUDA_ARCH_LIST = os.environ.get("TORCH_CUDA_ARCH_LIST")
 
+if TORCH_CUDA_ARCH_LIST:
+    arch_list = [arch for arch in TORCH_CUDA_ARCH_LIST.split() if float(arch.split('+')[0]) >= 6.0]
+    os.environ["TORCH_CUDA_ARCH_LIST"] = " ".join(arch_list)
+
 version_vars = {}
 exec("exec(open('gptqmodel/version.py').read()); version=__version__", {}, version_vars)
 gptqmodel_version = version_vars['version']
@@ -109,6 +113,7 @@ def get_version_tag(is_cuda_release: bool = True) -> str:
         if got_cuda_between_v6_and_v8:
             FORCE_BUILD = True
 
+
 if BUILD_CUDA_EXT:
     if CUDA_RELEASE == "1":
         common_setup_kwargs["version"] += f"+{get_version_tag(True)}"