From 9999058c889cf6367401f6b37e45a1b493ce81f3 Mon Sep 17 00:00:00 2001
From: lukasvwk <55445939+lukasvwk@users.noreply.github.com>
Date: Tue, 13 Aug 2024 14:51:22 +0200
Subject: [PATCH] Update setup_linux.py

---
 lib/nms/setup_linux.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/nms/setup_linux.py b/lib/nms/setup_linux.py
index 9120a93b..b78fa312 100644
--- a/lib/nms/setup_linux.py
+++ b/lib/nms/setup_linux.py
@@ -124,7 +124,7 @@ def build_extensions(self):
         # we're only going to use certain compiler args with nvcc and not with
         # gcc the implementation of this trick is in customize_compiler() below
         extra_compile_args={'gcc': ["-Wno-unused-function"],
-                            'nvcc': ['-arch=sm_35',
+                            'nvcc': ['-arch=compute_50',
                                      '--ptxas-options=-v',
                                      '-c',
                                      '--compiler-options',