docs: add instructions for cpu-only installation (#56)

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
This commit is contained in:
Michele Dolfi
2024-08-30 10:20:21 +02:00
committed by GitHub
parent 5c46749e70
commit a8a60d52b1
2 changed files with 21 additions and 1 deletions

View File

@@ -6,7 +6,10 @@ RUN apt-get update \
&& apt-get install -y libgl1 libglib2.0-0 curl wget git \
&& apt-get clean
RUN pip install --no-cache-dir docling
# This will install torch with *only* cpu support
# Remove the --extra-index-url part if you want to install all the gpu requirements
# For more details in the different torch distribution visit https://pytorch.org/.
RUN pip install --no-cache-dir docling --extra-index-url https://download.pytorch.org/whl/cpu
ENV HF_HOME=/tmp/
ENV TORCH_HOME=/tmp/