Skip to content

GitLab

Commit e3e09c46 authored by jnwei's avatar jnwei Committed by Jennifer Wei
Browse files

Remove nvcc compute capability 37 which caused kernel build issues

parent 4172f34b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
setup.py
View file @ e3e09c46
...@@ -52,7 +52,6 @@ def get_cuda_bare_metal_version(cuda_dir): ...@@ -52,7 +52,6 @@ def get_cuda_bare_metal_version(cuda_dir):
return raw_output, bare_metal_major, bare_metal_minor return raw_output, bare_metal_major, bare_metal_minor
compute_capabilities = set([ compute_capabilities = set([
(3, 7), # K80, e.g.
(5, 2), # Titan X (5, 2), # Titan X
(6, 1), # GeForce 1000-series (6, 1), # GeForce 1000-series
]) ])
...@@ -130,7 +129,7 @@ setup( ...@@ -130,7 +129,7 @@ setup(
classifiers=[ classifiers=[
'License :: OSI Approved :: Apache Software License', 'License :: OSI Approved :: Apache Software License',
'Operating System :: POSIX :: Linux', 'Operating System :: POSIX :: Linux',
'Programming Language :: Python :: 3.9,' 'Programming Language :: Python :: 3.10,'
'Topic :: Scientific/Engineering :: Artificial Intelligence', 'Topic :: Scientific/Engineering :: Artificial Intelligence',
], ],
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment