-
Notifications
You must be signed in to change notification settings - Fork 47
Description
pip install fairy-doc[gpu]
use DocConverter to convert a PDF(OCR) file.
`
failed to process ocr, reason: (PreconditionNotMet) Cannot load cudnn shared library. Cannot invoke method cudnnGetVersion.
[Hint: cudnn_dso_handle should not be null.] (at /paddle/paddle/phi/backends/dynload/cudnn.cc:64)
Traceback (most recent call last):
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/docconv.py", line 175, in _timeout_convert
res = func_timeout(
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/func_timeout/dafunc.py", line 108, in func_timeout
raise_exception(exception)
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/func_timeout/py3_raise.py", line 7, in raise_exception
raise exception[0] from None
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/conv/pdf_pp_structurev2.py", line 39, in to_md
model_list = model(bits)
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/conv/pdf_pp_structurev2.py", line 33, in call
return self.model(images) # type: ignore
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/model/doc_analysis_by_pp.py", line 12, in call
raise Exception("fatal error: failed to inference using paddleocr")
Exception: fatal error: failed to inference using paddleocr
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "/data/sdc/deploy/fairy-doc/benchmark.fairy-doc.cpu.py", line 32, in
[convert(args.path, file) for file in sorted(os.listdir(args.path)) if file.endswith(args.type)]
File "/data/sdc/deploy/fairy-doc/benchmark.fairy-doc.cpu.py", line 32, in
[convert(args.path, file) for file in sorted(os.listdir(args.path)) if file.endswith(args.type)]
File "/data/sdc/deploy/fairy-doc/benchmark.fairy-doc.cpu.py", line 14, in convert
markdown_content, time_cost = converter.convert(fn_in, conv_timeout=600)
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/docconv.py", line 203, in convert
return self._timeout_convert(
File "/home/ailearn/.conda/envs/fairy-doc/lib/python3.10/site-packages/magic_doc/docconv.py", line 190, in _timeout_convert
raise ConvException("Convert failed: %s" % str(ex))
magic_doc.docconv.ConvException: Convert failed: fatal error: failed to inference using paddleocr
`
NVIDIA-SMI 555.42.02 Driver Version: 555.42.02 CUDA Version: 12.5