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

確か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'
    
}

MKPのムジン社長

当サイトはリンクフリーです。
ご自身のブログでの引用、TwitterやFacebook、Instagram、Pinterestなどで当サイトの記事URLを共有していただくのは、むしろありがたいことです。
事前連絡や事後の連絡も不要ですが、ご連絡いただければ弊社も貴社のコンテンツを紹介させていただく可能性がございます。