TLDR-462 gpu for 1.1 - Githubissues

ispras / dedoc

Dedoc is a library (service) for automate documents parsing and bringing to a uniform format. It automatically extracts content, logical structure, tables, and meta information from textual electronic documents. (Parse document; Document content extraction; Logical structure extraction; PDF parser; Scanned document parser; DOCX parser; HTML parser

Apache License 2.0

111 stars 15 forks source link

TLDR-462 gpu for 1.1 #365

Closed raxtemur closed 8 months ago

raxtemur commented 8 months ago

Added gpu config and tested on XGBoost and torch models. Added in:

AbstractPickledLineTypeClassifier
TxtlayerClassifier
ScanParagraphClassifierExtractor
PdfImageReader

NastyBoget commented 8 months ago

Видимо еще нужно сделать try/except на ValueError, потому что если нет nvidia в принципе, то вылетает ошибка:

ValueError: invalid literal for int() with base 10: "NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running."