🚧 Support CUDA
parent
82d3d71ed4
commit
8eba477b20
@ -0,0 +1,51 @@
|
|||||||
|
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
|
||||||
|
|
||||||
|
ARG PIP_INSTALL="pip install --no-cache-dir --use-feature=2020-resolver"
|
||||||
|
ARG OCRD_VERSION_MINIMUM="2.18.1"
|
||||||
|
ENV LC_ALL=C.UTF-8 LANG=C.UTF-8
|
||||||
|
ENV PIP_DEFAULT_TIMEOUT=120
|
||||||
|
|
||||||
|
|
||||||
|
RUN echo "APT::Acquire::Retries \"3\";" > /etc/apt/apt.conf.d/80-retries && \
|
||||||
|
apt-get update && \
|
||||||
|
apt-get install -y \
|
||||||
|
curl xz-utils \
|
||||||
|
build-essential python3-dev \
|
||||||
|
# For get-pip.py:
|
||||||
|
python3-distutils \
|
||||||
|
# For add-apt-repository:
|
||||||
|
software-properties-common \
|
||||||
|
# XML utils
|
||||||
|
libxml2-utils \
|
||||||
|
xmlstarlet \
|
||||||
|
&& \
|
||||||
|
apt-get clean && \
|
||||||
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
|
||||||
|
# Set up OCR-D logging
|
||||||
|
RUN echo "setOverrideLogLevel(os.getenv('LOG_LEVEL', 'INFO'))" >/etc/ocrd_logging.py
|
||||||
|
|
||||||
|
|
||||||
|
# Install pip (and setuptools)
|
||||||
|
# We use get-pip.py here to avoid
|
||||||
|
# a. having to upgrade from Ubuntu's pip
|
||||||
|
# b. the dreaded "old script wrapper" error message
|
||||||
|
RUN curl -sSL https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
|
||||||
|
python3 get-pip.py && \
|
||||||
|
rm -f get-pip.py
|
||||||
|
|
||||||
|
|
||||||
|
# Install pip installable-stuff
|
||||||
|
RUN ${PIP_INSTALL} \
|
||||||
|
"ocrd >= ${OCRD_VERSION_MINIMUM}"
|
||||||
|
|
||||||
|
|
||||||
|
# Check pip dependencies
|
||||||
|
RUN pip check
|
||||||
|
|
||||||
|
|
||||||
|
WORKDIR /data
|
||||||
|
|
||||||
|
# Default command
|
||||||
|
CMD ['ocrd']
|
@ -0,0 +1,51 @@
|
|||||||
|
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
|
||||||
|
|
||||||
|
ARG PIP_INSTALL="pip install --no-cache-dir --use-feature=2020-resolver"
|
||||||
|
ARG OCRD_VERSION_MINIMUM="2.18.1"
|
||||||
|
ENV LC_ALL=C.UTF-8 LANG=C.UTF-8
|
||||||
|
ENV PIP_DEFAULT_TIMEOUT=120
|
||||||
|
|
||||||
|
|
||||||
|
RUN echo "APT::Acquire::Retries \"3\";" > /etc/apt/apt.conf.d/80-retries && \
|
||||||
|
apt-get update && \
|
||||||
|
apt-get install -y \
|
||||||
|
curl xz-utils \
|
||||||
|
build-essential python3-dev \
|
||||||
|
# For get-pip.py:
|
||||||
|
python3-distutils \
|
||||||
|
# For add-apt-repository:
|
||||||
|
software-properties-common \
|
||||||
|
# XML utils
|
||||||
|
libxml2-utils \
|
||||||
|
xmlstarlet \
|
||||||
|
&& \
|
||||||
|
apt-get clean && \
|
||||||
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
|
||||||
|
# Set up OCR-D logging
|
||||||
|
RUN echo "setOverrideLogLevel(os.getenv('LOG_LEVEL', 'INFO'))" >/etc/ocrd_logging.py
|
||||||
|
|
||||||
|
|
||||||
|
# Install pip (and setuptools)
|
||||||
|
# We use get-pip.py here to avoid
|
||||||
|
# a. having to upgrade from Ubuntu's pip
|
||||||
|
# b. the dreaded "old script wrapper" error message
|
||||||
|
RUN curl -sSL https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \
|
||||||
|
python3 get-pip.py && \
|
||||||
|
rm -f get-pip.py
|
||||||
|
|
||||||
|
|
||||||
|
# Install pip installable-stuff
|
||||||
|
RUN ${PIP_INSTALL} \
|
||||||
|
"ocrd >= ${OCRD_VERSION_MINIMUM}"
|
||||||
|
|
||||||
|
|
||||||
|
# Check pip dependencies
|
||||||
|
RUN pip check
|
||||||
|
|
||||||
|
|
||||||
|
WORKDIR /data
|
||||||
|
|
||||||
|
# Default command
|
||||||
|
CMD ['ocrd']
|
Loading…
Reference in New Issue