vuvanhung commited on
Commit
098a320
·
verified ·
1 Parent(s): 851768b

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +71 -0
Dockerfile ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.9-slim
2
+
3
+ ENV PYTHONUNBUFFERED=1
4
+ ENV PYTHONDONTWRITEBYTECODE=1
5
+ ENV PIP_NO_CACHE_DIR=1
6
+ ENV PIP_DISABLE_PIP_VERSION_CHECK=1
7
+
8
+ ENV HF_HOME=/tmp/huggingface_cache
9
+ ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache/transformers
10
+ ENV HF_DATASETS_CACHE=/tmp/huggingface_cache/datasets
11
+ ENV TORCH_HOME=/tmp/torch_cache
12
+ ENV TOKENIZERS_PARALLELISM=false
13
+
14
+ ENV OMP_NUM_THREADS=1
15
+ ENV MKL_NUM_THREADS=1
16
+ ENV OPENBLAS_NUM_THREADS=1
17
+
18
+ RUN apt-get update && apt-get install -y \
19
+ ffmpeg \
20
+ libavcodec-extra \
21
+ espeak-ng \
22
+ alsa-utils \
23
+ libasound2-dev \
24
+ libsndfile1 \
25
+ cmake \
26
+ build-essential \
27
+ pkg-config \
28
+ gcc \
29
+ g++ \
30
+ curl \
31
+ wget \
32
+ && apt-get clean \
33
+ && rm -rf /var/lib/apt/lists/* \
34
+ && rm -rf /tmp/* \
35
+ && rm -rf /var/tmp/*
36
+
37
+ RUN pip install --no-cache-dir sentencepiece
38
+
39
+ RUN mkdir -p $HF_HOME $TORCH_HOME && \
40
+ chmod -R 777 $HF_HOME $TORCH_HOME
41
+
42
+ WORKDIR /code
43
+
44
+ COPY requirements.txt /code/requirements.txt
45
+
46
+ RUN pip install --no-cache-dir sentencepiece\
47
+ pip install --no-cache-dir --upgrade pip setuptools wheel && \
48
+ pip install --no-cache-dir torch==2.0.1 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cpu && \
49
+
50
+ pip install --no-cache-dir --upgrade -r /code/requirements.txt && \
51
+
52
+ pip cache purge && \
53
+ rm -rf ~/.cache/pip && \
54
+ rm -rf /tmp/* && \
55
+ rm -rf /var/tmp/*
56
+
57
+ COPY . /code/
58
+
59
+ RUN useradd --create-home --shell /bin/bash app && \
60
+ chown -R app:app /code && \
61
+ chown -R app:app $HF_HOME && \
62
+ chown -R app:app $TORCH_HOME
63
+
64
+ USER app
65
+
66
+ HEALTHCHECK --interval=30s --timeout=30s --start-period=60s --retries=3 \
67
+ CMD curl -f http://localhost:7860/health || exit 1
68
+
69
+ EXPOSE 7860
70
+
71
+ CMD ["python", "-u", "app.py"]