This commit is contained in:
James Lamb 2024-09-02 23:12:35 -05:00
Родитель 90f80903a9
Коммит 231a6635d5
Не найден ключ, соответствующий данной подписи
6 изменённых файлов: 19 добавлений и 8 удалений

Просмотреть файл

@ -23,9 +23,8 @@ RUN curl -L -o cmake.sh https://github.com/Kitware/CMake/releases/download/v3.29
RUN git clone \ RUN git clone \
--recursive \ --recursive \
--branch ${LIGHTGBM_GIT_REF} \
--depth 1 \
https://github.com/Microsoft/LightGBM && \ https://github.com/Microsoft/LightGBM && \
git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd ./LightGBM && \ cd ./LightGBM && \
cmake -B build -S . && \ cmake -B build -S . && \
cmake --build build -j4 && \ cmake --build build -j4 && \

Просмотреть файл

@ -23,7 +23,10 @@ RUN apt-get update && \
conda config --set always_yes yes --set changeps1 no && \ conda config --set always_yes yes --set changeps1 no && \
# lightgbm # lightgbm
conda install -q -y numpy scipy scikit-learn pandas && \ conda install -q -y numpy scipy scikit-learn pandas && \
git clone --recursive --branch ${LIGHTGBM_GIT_REF} --depth 1 https://github.com/Microsoft/LightGBM && \ git clone \
--recursive \
https://github.com/Microsoft/LightGBM && \
git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd ./LightGBM && \ cd ./LightGBM && \
sh ./build-python.sh install && \ sh ./build-python.sh install && \
# clean # clean

Просмотреть файл

@ -9,8 +9,8 @@ RUN apt-get update && \
libomp-dev && \ libomp-dev && \
git clone \ git clone \
--recursive \ --recursive \
--branch ${LIGHTGBM_GIT_REF} \ https://github.com/Microsoft/LightGBM && \
--depth 1 https://github.com/Microsoft/LightGBM && \ git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd ./LightGBM && \ cd ./LightGBM && \
sh build-cran-package.sh --no-build-vignettes && \ sh build-cran-package.sh --no-build-vignettes && \
MAKEFLAGS="-j2" R CMD INSTALL ./lightgbm_*.tar.gz && \ MAKEFLAGS="-j2" R CMD INSTALL ./lightgbm_*.tar.gz && \

Просмотреть файл

@ -40,7 +40,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# LightGBM # LightGBM
WORKDIR /opt WORKDIR /opt
RUN git clone --recursive --branch ${LIGHTGBM_GIT_REF} --depth 1 https://github.com/Microsoft/LightGBM && \ RUN git clone \
--recursive \
https://github.com/Microsoft/LightGBM && \
git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd LightGBM && \ cd LightGBM && \
cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=${OPENCL_LIBRARIES}/libOpenCL.so.1 -DOpenCL_INCLUDE_DIR=$OPENCL_INCLUDE_DIR && \ cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=${OPENCL_LIBRARIES}/libOpenCL.so.1 -DOpenCL_INCLUDE_DIR=$OPENCL_INCLUDE_DIR && \
OPENCL_HEADERS=$OPENCL_INCLUDE_DIR LIBOPENCL=$OPENCL_LIBRARIES cmake --build build OPENCL_HEADERS=$OPENCL_INCLUDE_DIR LIBOPENCL=$OPENCL_LIBRARIES cmake --build build

Просмотреть файл

@ -40,7 +40,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
# LightGBM # LightGBM
WORKDIR /opt WORKDIR /opt
RUN git clone --recursive --branch ${LIGHTGBM_GIT_REF} --depth 1 https://github.com/Microsoft/LightGBM && \ RUN git clone \
--recursive \
https://github.com/Microsoft/LightGBM && \
git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd LightGBM && \ cd LightGBM && \
cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=${OPENCL_LIBRARIES}/libOpenCL.so.1 -DOpenCL_INCLUDE_DIR=$OPENCL_INCLUDE_DIR && \ cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=${OPENCL_LIBRARIES}/libOpenCL.so.1 -DOpenCL_INCLUDE_DIR=$OPENCL_INCLUDE_DIR && \
OPENCL_HEADERS=$OPENCL_INCLUDE_DIR LIBOPENCL=$OPENCL_LIBRARIES cmake --build build OPENCL_HEADERS=$OPENCL_INCLUDE_DIR LIBOPENCL=$OPENCL_LIBRARIES cmake --build build

Просмотреть файл

@ -82,7 +82,10 @@ RUN conda config --set always_yes yes --set changeps1 no && \
################################################################################################################# #################################################################################################################
RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \ RUN cd /usr/local/src && mkdir lightgbm && cd lightgbm && \
git clone --recursive --branch ${LIGHTGBM_GIT_REF} --depth 1 https://github.com/microsoft/LightGBM && \ git clone \
--recursive \
https://github.com/Microsoft/LightGBM && \
git -C ./LightGBM checkout ${LIGHTGBM_GIT_REF} && \
cd LightGBM && \ cd LightGBM && \
cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ && \ cmake -B build -S . -DUSE_GPU=1 -DOpenCL_LIBRARY=/usr/local/cuda/lib64/libOpenCL.so -DOpenCL_INCLUDE_DIR=/usr/local/cuda/include/ && \
OPENCL_HEADERS=/usr/local/cuda-8.0/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-8.0/targets/x86_64-linux/lib cmake --build build OPENCL_HEADERS=/usr/local/cuda-8.0/targets/x86_64-linux/include LIBOPENCL=/usr/local/cuda-8.0/targets/x86_64-linux/lib cmake --build build