|
FROM python:3.8 |
|
LABEL maintainer="Hugging Face" |
|
|
|
RUN apt update |
|
RUN git clone https: |
|
|
|
RUN python3 -m pip install --no-cache-dir --upgrade pip && python3 -m pip install --no-cache-dir git+https: |
|
RUN apt-get -y update && apt-get install -y libsndfile1-dev && apt install -y tesseract-ocr |
|
|
|
# Torch needs to be installed before deepspeed |
|
RUN python3 -m pip install --no-cache-dir ./transformers[deepspeed] |
|
|
|
RUN python3 -m pip install --no-cache-dir torchvision git+https: |
|
RUN python3 -m pip install --no-cache-dir pytorch-quantization --extra-index-url https: |
|
RUN python3 -m pip install -U "itsdangerous<2.1.0" |
|
|
|
# Test if the image could successfully build the doc. before publishing the image |
|
RUN doc-builder build transformers transformers/docs/source/en --build_dir doc-build-dev --notebook_dir notebooks/transformers_doc --clean |
|
RUN rm -rf doc-build-dev |