Add files via upload

This commit is contained in:
Lectom
2024-12-06 11:05:05 +09:00
committed by GitHub
parent 97a19fe647
commit 308857344c
6 changed files with 253 additions and 0 deletions

26
dockerfile Normal file
View File

@@ -0,0 +1,26 @@
# FROM huggingface/transformers-pytorch-gpu
FROM pytorch/pytorch:latest
ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Asia/Seoul
ADD ./workspace /opt/workspace
WORKDIR /opt/workspace
# Latest version of transformers-pytorch-gpu seems to lack tk.
# Further, pip install fails, so we must upgrade pip first.
RUN cd /etc/apt && \
sed -i 's/archive.ubuntu.com/mirror.kakao.com/g' sources.list
RUN apt-get -y update
RUN apt-get -y install python3-tk git vim curl
RUN python3 -m pip install --upgrade pip
COPY requirements.txt .
RUN python3 -m pip install -r requirements.txt
# To be able to see logs in real time
ENV PYTHONUNBUFFERED=1
# ENTRYPOINT ["python3"]
# CMD ["app.py", "--server_name", "0.0.0.0", "--auto_parallel", "True"]