Modifying arch compilation command in setup to work with Titan X

wzheng1983 · Feb 13, 2017 · 66d36f0 · 66d36f0
1 parent b1077ef
commit 66d36f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/setup.py b/lib/setup.py
@@ -131,7 +131,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_61',
+                            'nvcc': ['-arch=sm_52',
                                      '--ptxas-options=-v',
                                      '-c',
                                      '--compiler-options',