Update __init__.py

k2-fsa · Jul 6, 2023 · e94cc52 · e94cc52
1 parent 8cff04a
commit e94cc52
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/k2/python/k2/__init__.py b/k2/python/k2/__init__.py
@@ -10,16 +10,6 @@
         f"But you are using PyTorch {torch.__version__} to run it"
     )
 
-if (
-    k2_torch_cuda_version != ""
-    and torch.version.cuda is not None
-    and torch.version.cuda != k2_torch_cuda_version
-):
-    raise ImportError(
-        f"k2 was built using CUDA {k2_torch_cuda_version}\n"
-        f"But you are using CUDA {torch.version.cuda} to run it."
-    )
-
 from _k2 import DeterminizeWeightPushingType
 from _k2 import simple_ragged_index_select
 from _k2 import swoosh_l