aditya/llama.cpp
1
0
Fork 0
mirror of https://git.adityakumar.xyz/llama.cpp.git synced 2024-11-09 15:29:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove torch GPU dependencies from the Docker.full image (#665)

Browse source 
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
This commit is contained in:
bsilvereagle 2023-04-02 15:13:03 -07:00 committed by GitHub
parent d8d4e865cd
commit a0c0516416
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
.devops/full.Dockerfile
View file

@ -6,7 +6,8 @@ RUN apt-get update && \
apt-get install -y build-essential python3 python3-pip apt-get install -y build-essential python3 python3-pip
RUN pip install --upgrade pip setuptools wheel \ 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 WORKDIR /app