【Python】GPU認識の確認とGPU情報の表示 | 合同会社ムジンケイカクプロ

【Python】GPU認識の確認とGPU情報の表示

python Python

確かLightgbmでGPU利用をしようとした時に、GPUを認識しているかの確認に使ったはずです。

from tensorflow.python.client import device_lib
device_lib.list_local_devices()
# LightGBM のハイパーパラメータ
lgbm_params = {

    # 多値分類問題
    'objective': 'multiclass',

    # クラス数は 3
    'num_class': 3,

    'device':'gpu'
    
}

タイトルとURLをコピーしました