We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6f69e46 commit 9999058Copy full SHA for 9999058
lib/nms/setup_linux.py
@@ -124,7 +124,7 @@ def build_extensions(self):
124
# we're only going to use certain compiler args with nvcc and not with
125
# gcc the implementation of this trick is in customize_compiler() below
126
extra_compile_args={'gcc': ["-Wno-unused-function"],
127
- 'nvcc': ['-arch=sm_35',
+ 'nvcc': ['-arch=compute_50',
128
'--ptxas-options=-v',
129
'-c',
130
'--compiler-options',
0 commit comments