반응형
Kaggle Notebook에선 lightGBM gpu 버전을 쉽게 사용할 수 있다. 하지만 Google CoLab 에선 별도로 설치가 필요하다. 아래 코드를 CoLab 노트북 셀에서 실행하여 설치한 후 사용해본다.
!git clone --recursive https://github.com/Microsoft/LightGBM
%cd LightGBM
!mkdir build
%cd build
!cmake ../../LightGBM -DUSE_GPU=1
!make -j4
%cd ../python-package
!python3 setup.py install --gpu
반응형
'AI > Tips' 카테고리의 다른 글
학습률 웜업(Learning Rate WarmUp)에 대한 이해 (0) | 2024.05.06 |
---|---|
[Numpy] Serise clip() 함수 - 값 범위 제한하기 (0) | 2024.04.19 |
[XGBoost] XGBoost 학습 시 GPU(CUDA) 사용하기 (0) | 2024.03.31 |