Skip to content

Commit a0c0516

Browse files
authored
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
1 parent d8d4e86 commit a0c0516

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.devops/full.Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@ RUN apt-get update && \
66
apt-get install -y build-essential python3 python3-pip
77

88
RUN pip install --upgrade pip setuptools wheel \
9-
&& pip install numpy requests sentencepiece torch tqdm
9+
&& pip install numpy requests sentencepiece tqdm \
10+
&& pip install torch --index-url https://download.pytorch.org/whl/cpu
1011

1112
WORKDIR /app
1213

0 commit comments

Comments
 (0)