File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -71,7 +71,7 @@ RUN apt-get install -y libfreetype6-dev && \
71
71
pip install wordcloud && \
72
72
pip install xgboost && \
73
73
# Pinned to match GPU version. Update version together.
74
- pip install lightgbm==3.2.0 && \
74
+ pip install lightgbm==3.2.1 && \
75
75
pip install pydot && \
76
76
pip install keras && \
77
77
pip install keras-tuner && \
Original file line number Diff line number Diff line change @@ -67,7 +67,7 @@ RUN pip uninstall -y lightgbm && \
67
67
cd /usr/local/src && \
68
68
git clone --recursive https://github.com/microsoft/LightGBM && \
69
69
cd LightGBM && \
70
- git checkout tags/v3.2.0 && \
70
+ git checkout tags/v3.2.1 && \
71
71
mkdir build && cd build && \
72
72
cmake -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ .. && \
73
73
make -j$(nproc) && \
You can’t perform that action at this time.
0 commit comments