🚧 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