LightGBM GPU Support

Hi, I want to make use of LightGBM with the GPUs on Swan. However, I’m having trouble setting that up on swan as I’m unaware of the Cuda backend configs. Would appreciate any support/instructions to get it working.

Dear Yash,

What libraries do you need for LightGBM to work?

The LCG CUDA stacks (LCG 105a CUDA is the latest for now) include the CUDA toolkit, and the NVIDIA driver also mounts some key libraries (e.g. libnvidia-opencl).

The best would be for you to create a SWAN session with a GPU, try to install the lightgbm Python package (if that’s what you want) following the instructions here:

and then check for possibly missing libraries.