mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-07-11 22:29:29 +02:00
Docker: update to ONNX base image
This commit is contained in:
parent
171a8a3161
commit
32568a590f
1 changed files with 3 additions and 3 deletions
|
|
@ -15,7 +15,7 @@ LABEL \
|
|||
org.opencontainers.image.documentation="https://github.com/qurator-spk/eynollah/blob/${VCS_REF}/README.md" \
|
||||
org.opencontainers.image.revision=$VCS_REF \
|
||||
org.opencontainers.image.created=$BUILD_DATE \
|
||||
org.opencontainers.image.base.name=ocrd/core-cuda-tf2
|
||||
org.opencontainers.image.base.name=ocrd/core-cuda-onnx
|
||||
|
||||
ENV DEBIAN_FRONTEND=noninteractive
|
||||
# set proper locales
|
||||
|
|
@ -40,8 +40,8 @@ RUN ocrd ocrd-tool ocrd-tool.json dump-tools > $(dirname $(ocrd bashlib filename
|
|||
RUN ocrd ocrd-tool ocrd-tool.json dump-module-dirs > $(dirname $(ocrd bashlib filename))/ocrd-all-module-dir.json
|
||||
# install everything and reduce image size
|
||||
RUN make install EXTRAS=OCR && rm -rf /build/eynollah
|
||||
# fixup for broken cuDNN installation (Torch pulls in 8.5.0, which is incompatible with Tensorflow)
|
||||
RUN pip install nvidia-cudnn-cu11==8.6.0.163
|
||||
# fixup for broken cuDNN installation (Torch may pull in version which is incompatible with Tensorflow)
|
||||
# but most recent Torch versions pull cu13 variants, which are not conflicting here
|
||||
# smoke test
|
||||
RUN eynollah --help
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue