From 263040e6d90b483ddbcffdea04f307275f830be3 Mon Sep 17 00:00:00 2001 From: naruto110 Date: Wed, 27 Aug 2025 19:47:28 +0800 Subject: [PATCH] Update setup.py To adapt to the RTX 5090, we have modified the **setup.py** file. --- setup.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/setup.py b/setup.py index 32d62ed9..bb08396d 100755 --- a/setup.py +++ b/setup.py @@ -191,8 +191,6 @@ def append_nvcc_threads(nvcc_extra_args): cc_flag.append("-gencode") cc_flag.append("arch=compute_90,code=sm_90") if bare_metal_version >= Version("12.8"): - cc_flag.append("-gencode") - cc_flag.append("arch=compute_100,code=sm_100") cc_flag.append("-gencode") cc_flag.append("arch=compute_120,code=sm_120")