File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff line change @@ -25,11 +25,9 @@ RUN pip uninstall -y horovod && \
25
25
# as described by Lionel Chan at http://stackoverflow.com/a/37426929/5881346
26
26
RUN sed -i "s/httpredir.debian.org/debian.uchicago.edu/" /etc/apt/sources.list && \
27
27
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)
30
29
# 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 && \
33
31
# b/182601974: ssh client was removed from the base image but is required for packages such as stable-baselines.
34
32
apt-get install -y openssh-client && \
35
33
/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
64
62
# Install LightGBM
65
63
ENV LIGHTGBM_VERSION=3.2.1
66
64
{{ 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 && \
69
67
mkdir -p /etc/OpenCL/vendors && \
70
68
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd && \
71
69
cd /usr/local/src && \
You can’t perform that action at this time.
0 commit comments