Skip to content

Commit 7cf514a

Browse files
authored
Remove unnecessary packages (#1082)
1 parent 81896a5 commit 7cf514a

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

Dockerfile.tmpl

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,9 @@ RUN pip uninstall -y horovod && \
2525
# as described by Lionel Chan at http://stackoverflow.com/a/37426929/5881346
2626
RUN sed -i "s/httpredir.debian.org/debian.uchicago.edu/" /etc/apt/sources.list && \
2727
apt-get update && \
28-
# Needed by vowpalwabbit & lightGBM (GPU build).
29-
# https://github.com/VowpalWabbit/vowpal_wabbit/wiki/Python#installing
28+
# Needed by lightGBM (GPU build)
3029
# https://lightgbm.readthedocs.io/en/latest/GPU-Tutorial.html#build-lightgbm
31-
apt-get install -y build-essential unzip cmake && \
32-
apt-get install -y libboost-dev libboost-program-options-dev libboost-system-dev libboost-thread-dev libboost-math-dev libboost-test-dev libboost-python-dev libboost-filesystem-dev zlib1g-dev && \
30+
apt-get install -y build-essential unzip cmake libboost-dev libboost-system-dev libboost-filesystem-dev && \
3331
# b/182601974: ssh client was removed from the base image but is required for packages such as stable-baselines.
3432
apt-get install -y openssh-client && \
3533
/tmp/clean-layer.sh
@@ -64,8 +62,8 @@ RUN pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 torcht
6462
# Install LightGBM
6563
ENV LIGHTGBM_VERSION=3.2.1
6664
{{ if eq .Accelerator "gpu" }}
67-
# Install OpenCL & libboost (required by LightGBM GPU version)
68-
RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \
65+
# Install OpenCL (required by LightGBM GPU version)
66+
RUN apt-get install -y ocl-icd-libopencl1 clinfo && \
6967
mkdir -p /etc/OpenCL/vendors && \
7068
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd && \
7169
cd /usr/local/src && \

0 commit comments

Comments
 (0)