Display cuda version for module

Author: atalman

test/smoke_test/smoke_test.py

@@ -57,18 +57,6 @@ def check_nightly_binaries_date(package: str) -> None:
                     f"Expected {module['name']} to be less then {NIGHTLY_ALLOWED_DELTA} days. But its {date_m_delta}"
                 )
 
-def check_cuda_version(version: str, dlibary: str):
-    if version is not None and torch.version.cuda is not None:
-        version_str = str(version)
-        m_version = f"{version_str[:-3]}.{version_str[-2]}"
-        t_version = torch.version.cuda.split(".")
-        t_version = f"{t_version[0]}.{t_version[1]}"
-        if m_version != t_version:
-            raise RuntimeError(
-                "Detected that PyTorch and {dlibary} were compiled with different CUDA versions. "
-                f"PyTorch has CUDA version {t_version} whereas {dlibary} has CUDA version {m_version}. "
-            )
-
 def smoke_test_cuda(package: str) -> None:
     if not torch.cuda.is_available() and is_cuda_system:
         raise RuntimeError(f"Expected CUDA {gpu_arch_ver}. However CUDA is not loaded.")
@@ -86,7 +74,8 @@ def smoke_test_cuda(package: str) -> None:
         for module in MODULES:
             imported_module = importlib.import_module(module["name"])
             if module["extension"] == "extension":
-                imported_module.extension._check_cuda_version()
+                version = imported_module.extension._check_cuda_version()
+                print(f"{module['name']} CUDA: {version}")
             else:
                 imported_module._extension._check_cuda_version()