chendl/compositional_test

1FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.042LABEL maintainer="Hugging Face"3 4ARG DEBIAN_FRONTEND=noninteractive5 6RUN apt update7RUN apt install -y git libsndfile1-dev tesseract-ocr espeak-ng python3 python3-pip ffmpeg8RUN python3 -m pip install --no-cache-dir --upgrade pip9 10ARG REF=main11RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF12RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing,video]13 14# If set to nothing, will install the latest version15ARG PYTORCH='2.0.0'16ARG TORCH_VISION=''17ARG TORCH_AUDIO=''18# Example: `cu102`, `cu113`, etc.19ARG CUDA='cu117'20 21RUN [ ${#PYTORCH} -gt 0 ] && VERSION='torch=='$PYTORCH'.*' ||  VERSION='torch'; python3 -m pip install --no-cache-dir -U $VERSION --extra-index-url https://download.pytorch.org/whl/$CUDA22RUN [ ${#TORCH_VISION} -gt 0 ] && VERSION='torchvision=='TORCH_VISION'.*' ||  VERSION='torchvision'; python3 -m pip install --no-cache-dir -U $VERSION --extra-index-url https://download.pytorch.org/whl/$CUDA23RUN [ ${#TORCH_AUDIO} -gt 0 ] && VERSION='torchaudio=='TORCH_AUDIO'.*' ||  VERSION='torchaudio'; python3 -m pip install --no-cache-dir -U $VERSION --extra-index-url https://download.pytorch.org/whl/$CUDA24 25RUN python3 -m pip uninstall -y tensorflow flax26 27RUN python3 -m pip install --no-cache-dir git+https://github.com/facebookresearch/detectron2.git pytesseract28RUN python3 -m pip install -U "itsdangerous<2.1.0"29 30# When installing in editable mode, `transformers` is not recognized as a package.31# this line must be added in order for python to be aware of transformers.32RUN cd transformers && python3 setup.py develop33