Error importing Detection model on Kaggle Notebooks: Torch and Cuda incompatibility

[iaminebriki]

I have installed surya-ocr with pip on Kaggle notebook and I'm getting this error when I try to import the Detection model which I suppose has something to do with Torch version?

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[11], line 2
      1 # ---------- detection ----------
----> 2 from surya.model.detection.model import load_model as load_det_model, load_processor as load_det_processor
      3 # ---------- layout ----------
      4 # from surya.detection import batch_text_detection
      5 # from surya.layout import batch_layout_detection
   (...)
     10 # from surya.model.ordering.processor import load_processor as ord_processor
     11 # ---------- ocr ----------
     12 from surya.ocr import run_ocr

File /opt/conda/lib/python3.10/site-packages/surya/model/detection/model.py:14
     11 from typing import Optional, Union, Tuple, List, Any
     12 from functools import partial
---> 14 import torch
     15 import torch.nn as nn
     16 import torch.nn.functional as F

File /opt/conda/lib/python3.10/site-packages/torch/__init__.py:367
    365     if USE_GLOBAL_DEPS:
    366         _load_global_deps()
--> 367     from torch._C import *  # noqa: F403
    370 class SymInt:
    371     """
    372     Like an int (including magic methods), but redirects all operations on the
    373     wrapped node. This is used in particular to symbolically record operations
    374     in the symbolic shape workflow.
    375     """

ImportError: /opt/conda/lib/python3.10/site-packages/torch/lib/../../nvidia/cusparse/lib/libcusparse.so.12: undefined symbol: __nvJitLinkComplete_12_4, version libnvJitLink.so.12

[iaminebriki]

For anyone faced with same issue, here is a workaround:

1. Install Surya in a separate env:

!mkdir my_env
!pip install --target=my_env surya-ocr

2. Add that env to PATH:

import sys
sys.path.append("my_env")
import surya