change default -arch in setup.py to sm_35 to prevent some compatibili…

…ty issue
PanUnsw · Aug 7, 2017 · e0ab1d4 · e0ab1d4
1 parent 0ba447b
commit e0ab1d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/setup.py b/lib/setup.py
@@ -134,7 +134,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_35',
                                      '--ptxas-options=-v',
                                      '-c',
                                      '--compiler-options',