diff --git a/op_builder/builder.py b/op_builder/builder.py index ac45c4805d558eb637b224e18016a73f6546f76d..f44aee79637ac188dfb7f4cd08db1b8b06778793 100644 --- a/op_builder/builder.py +++ b/op_builder/builder.py @@ -34,7 +34,7 @@ def installed_cuda_version(): def get_default_compute_capatabilities(): compute_caps = DEFAULT_COMPUTE_CAPABILITIES if installed_cuda_version()[0] >= 11: - compute_caps += ";8.0" + compute_caps += ";8.0;8.6" return compute_caps