diff --git a/setup.py b/setup.py index e3811f80..ff6c059c 100644 --- a/setup.py +++ b/setup.py @@ -24,7 +24,6 @@ def cuda_extension(): cuda_version = float(torch.version.cuda) nvcc_args = [ - "-gencode=arch=compute_70,code=sm_70", "-gencode=arch=compute_75,code=sm_75", ] if cuda_version >= 11: