RuntimeError: CUDA error: the launch timed out and was terminated

[yuisunn]

Sometimes this error will be reported torch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1
li_txts = self.ocr.readtext(img_arry)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\anaconda3\Lib\site-packages\easyocr\easyocr.py", line 452, in readtext
horizontal_list, free_list = self.detect(img,
^^^^^^^^^^^^^^^^
File "C:\Users\anaconda3\Lib\site-packages\easyocr\easyocr.py", line 321, in detect
text_box_list = self.get_textbox(self.detector,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\anaconda3\Lib\site-packages\easyocr\detection.py", line 95, in get_textbox
bboxes_list, polys_list = test_net(canvas_size, mag_ratio, detector,
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "C:\Users\anaconda3\Lib\site-packages\easyocr\detection.py", line 51, in test_net
score_text = out[:, :, 0].cpu().data.numpy()
^^^^^^^^^^^^^^^^^^
RuntimeError: CUDA error: the launch timed out and was terminated
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

[samthakur587]

use this

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch