diff options
Diffstat (limited to '.devops/full.Dockerfile')
| -rw-r--r-- | .devops/full.Dockerfile | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/.devops/full.Dockerfile b/.devops/full.Dockerfile index 2b3a20c..a75bc97 100644 --- a/.devops/full.Dockerfile +++ b/.devops/full.Dockerfile @@ -6,7 +6,8 @@ RUN apt-get update && \ apt-get install -y build-essential python3 python3-pip RUN pip install --upgrade pip setuptools wheel \ - && pip install numpy requests sentencepiece torch tqdm + && pip install numpy requests sentencepiece tqdm \ + && pip install torch --index-url https://download.pytorch.org/whl/cpu WORKDIR /app |