diff options
| author | bsilvereagle <bsilvereagle@users.noreply.github.com> | 2023-04-02 15:13:03 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-03 00:13:03 +0200 |
| commit | a0c05164168297c04737936ad0cad849a512547a (patch) | |
| tree | 30f6e731817e123d4402f95091aa3e6d320345c8 /.devops/full.Dockerfile | |
| parent | d8d4e865cd481b18f10508ffee35db903767ef5c (diff) | |
Remove torch GPU dependencies from the Docker.full image (#665)
By using `pip install torch --index-url https://download.pytorch.org/whl/cpu`
instead of `pip install torch` we can specify we want to install a CPU-only version
of PyTorch without any GPU dependencies. This reduces the size of the Docker image
from 7.32 GB to 1.62 GB
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 |