We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fa70484 commit bc19808Copy full SHA for bc19808
op_builder/builder.py
@@ -63,7 +63,7 @@ def get_default_compute_capabilities():
63
compute_caps = DEFAULT_COMPUTE_CAPABILITIES
64
# Update compute capability according to: https://en.wikipedia.org/wiki/CUDA#GPUs_supported
65
import torch.utils.cpp_extension
66
- if torch.utils.cpp_extension.CUDA_HOME is not None:
+ if torch.utils.cpp_extension.CUDA_HOME is not None:
67
if installed_cuda_version()[0] == 11:
68
if installed_cuda_version()[1] >= 0:
69
compute_caps += ";8.0"
0 commit comments