|
@@ -5,9 +5,10 @@ FROM ubuntu:$UBUNTU_VERSION as build
|
|
|
RUN apt-get update && \
|
|
RUN apt-get update && \
|
|
|
apt-get install -y build-essential python3 python3-pip
|
|
apt-get install -y build-essential python3 python3-pip
|
|
|
|
|
|
|
|
|
|
+COPY requirements.txt requirements.txt
|
|
|
|
|
+
|
|
|
RUN pip install --upgrade pip setuptools wheel \
|
|
RUN pip install --upgrade pip setuptools wheel \
|
|
|
- && pip install numpy requests sentencepiece tqdm \
|
|
|
|
|
- && pip install torch --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
|
|
|
+ && pip install -r requirements.txt
|
|
|
|
|
|
|
|
WORKDIR /app
|
|
WORKDIR /app
|
|
|
|
|
|