DA-Transformer / Dockerfile
hzhwcmhf's picture
fix env
552c70e
raw
history blame
1.48 kB
FROM python:3.9
RUN --mount=target=/root/packages.txt,source=packages.txt sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && sed -i 's http://archive.ubuntu.com http://us-east-1.ec2.archive.ubuntu.com g' /etc/apt/sources.list && sed -i '/security/d' /etc/apt/sources.list && apt-get update && xargs -r -a /root/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
RUN pip install --no-cache-dir pip==22.3.1 && pip install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1"
WORKDIR /home/user/app
RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && sed -i 's http://archive.ubuntu.com http://us-east-1.ec2.archive.ubuntu.com g' /etc/apt/sources.list && sed -i '/security/d' /etc/apt/sources.list && apt-get update && apt-get install -y git git-lfs ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx && rm -rf /var/lib/apt/lists/* && git lfs install
RUN pip install --no-cache-dir Cython "gradio==3.28.3" "torch==1.10.1"
RUN git clone --recurse-submodules https://github.com/thu-coai/DA-Transformer.git && cd DA-Transformer && git checkout demo && pip install -e . && cd dag_search && python3 setup.py build_ext --inplace && pip install -e . && cd ../..
RUN mkdir -p /home/user && chmod 777 /home/user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
COPY . .
CMD ["python3", "app.py"]