diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl
index 31c96caf5bd..31539ca9bf4 100644
--- a/third_party/gpus/cuda_configure.bzl
+++ b/third_party/gpus/cuda_configure.bzl
@@ -303,7 +303,7 @@ def compute_capabilities(repository_ctx):
         for prefix in ["compute_", "sm_"]:
             if not capability.startswith(prefix):
                 continue
-            if len(capability) == len(prefix) + 2 and capability[-2:].isdigit():
+            if len(capability) <= len(prefix) + 3 and capability[len(prefix):].isdigit():
                 continue
             auto_configure_fail("Invalid compute capability: %s" % capability)