Spaces:
Running
Running
File size: 1,040 Bytes
3d44d9f 31bb33f 3d44d9f 6b44420 3d44d9f ffde7ba 3d44d9f eb6ec9b c37d135 eb6ec9b 3d44d9f ebf61c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
FROM docker.io/nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
ENV TZ=Asia/Tokyo
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
RUN apt-get update && \
apt-get install -y \
git \
make build-essential libssl-dev zlib1g-dev \
libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm \
libncursesw5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev git-lfs \
python3.10-dev python3-pip && \
rm -rf /var/lib/apt/lists/* && \
git lfs install
RUN useradd -m -u 1000 user
USER user
COPY --link --chown=1000 ./requirements.txt /home/user/app/requirements.txt
WORKDIR /home/user/app
RUN pip install -U setuptools wheel
RUN pip install torch==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
RUN pip install -r requirements.txt
COPY --link --chown=1000 ./model_pull.py /home/user/app/model_pull.py
RUN python3.10 model_pull.py
COPY --link --chown=1000 ./app.py /home/user/app/app.py
EXPOSE 7860
ENTRYPOINT ["python3.10", "app.py"] |