diff --git a/third_party/gpus/find_cuda_config.py b/third_party/gpus/find_cuda_config.py
index 82d4ee16733..d948593e518 100644
--- a/third_party/gpus/find_cuda_config.py
+++ b/third_party/gpus/find_cuda_config.py
@@ -451,7 +451,7 @@ def find_cuda_config():
 
     cuda_version = result["cuda_version"]
     cublas_paths = base_paths
-    if cuda_version.split(".") < (10, 1):
+    if tuple(int(v) for v in cuda_version.split(".")) < (10, 1):
       # Before CUDA 10.1, cuBLAS was in the same directory as the toolkit.
       cublas_paths = cuda_paths
     cublas_version = os.environ.get("TF_CUBLAS_VERSION", "")