実現したいこと
タイトル通りですが
仮想環境下でGPU版のLightGBMをインストールしたい
前提
下記公式サイト通りにGPU版のlightGBMをビルドすることができました、ビルドしたGPU版LightGBMのインストールの仕方がわかりません。
https://lightgbm.readthedocs.io/en/latest/Installation-Guide.html#id18
git clone --recursive https://github.com/microsoft/LightGBM
cd LightGBM
cmake -B build -S . -A x64 -DUSE_GPU=1 -DBOOST_ROOT=C:/local/boost_1_63_0 -DBOOST_LIBRARYDIR=C:/local/boost_1_63_0/lib64-msvc-14.0
if you have installed NVIDIA CUDA to a customized location, you should specify paths to OpenCL headers and library like the following:
cmake -B build -S . -A x64 -DUSE_GPU=1 -DBOOST_ROOT=C:/local/boost_1_63_0 -DBOOST_LIBRARYDIR=C:/local/boost_1_63_0/lib64-msvc-14.0 -DOpenCL_LIBRARY="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.0/lib/x64/OpenCL.lib" -DOpenCL_INCLUDE_DIR="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.0/include"
cmake --build build --target ALL_BUILD --config Release
試したこと
下記コマンドを仮想環境下のコマンドプロンプト上で試しました
pip install lightgbm --install-option=--gpu
pip install lightgbm --install-option=--gpu --install-option="--boost-root=c:/boost_1_81_0" --install-option="--boost-librarydir=c:/boost_1_81_0/stage/lib/x64"
補足情報(FW/ツールのバージョンなど)
boost_1_81_0
lib64-msvc-14.3
NVIDIA GPU Computing Toolkit\CUDA\v12.1\lib\x64/OpenCL.lib
cmake version 3.29.3
Visual Studio 2022 version 17.9.7
0 コメント