mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-15 10:38:23 +06:00
78a471ff71
* First draft * Remove scatter dependency * Add require_torch * update vectorized sum test, add clone call * remove artifacts * fix style * fix style v2 * remove "scatter" mentions from the code base * fix isort error Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local> Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
19 lines
1023 B
Docker
19 lines
1023 B
Docker
FROM python:3.8
|
|
LABEL maintainer="Hugging Face"
|
|
|
|
RUN apt update
|
|
RUN git clone https://github.com/huggingface/transformers
|
|
|
|
RUN python3 -m pip install --no-cache-dir --upgrade pip && python3 -m pip install --no-cache-dir git+https://github.com/huggingface/doc-builder ./transformers[dev]
|
|
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://github.com/facebookresearch/detectron2.git pytesseract
|
|
RUN python3 -m pip install --no-cache-dir pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com
|
|
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 |