Colab GPU issue

[akshayrakate]

Sometimes when is try to run code with , I get an error as follows:

Code:

!pip install "python-doctr[tf]"
!pip install rapidfuzz==2.15.1
!pip install Pillow==9.5.0

from doctr.io import DocumentFile
from doctr.models import ocr_predictor

file_path = r"/content/image_file.png" 
img = DocumentFile.from_images([file_path])
result = predictor(img)

output:

InvalidArgumentError: {{function_node __wrapped__StridedSlice_device_/job:localhost/replica:0/task:0/device:GPU:0}} Expected begin, end, and strides to be 1D equal size tensors, but got shapes [2], [1], and [1] instead. [Op:StridedSlice] name: strided_slice/

[felixT2K]

Hi @akshayrakate 👋🏼 ,

Could you please try to install docTR from the main branch and test it again ?
We will do next week a release which includes some bug fixes

[akshayrakate]

Hi @felixT2K,
I solved that issue by disabling multiprocessing threading providing os.environ['DOCTR_MULTIPROCESSING_DISABLE'] = 'True'.