vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Remove torch GPU dependencies

Browse source
This commit is contained in:
bsilvereagle 2023-03-31 15:51:41 -07:00
parent 3525899277
commit 6a498f0d79
1 changed files with 2 additions and 1 deletions
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
RUN pip install --upgrade pip setuptools wheel \
&& pip install numpy requests sentencepiece torch tqdm
&& pip install numpy requests sentencepiece tqdm \
&& pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
WORKDIR /app