【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'
    
}

pip install tensorflow

tensorflow-gpuをインストールする時に、Python本体のバージョンとの互換性をチェックします。

※備忘と参考程度のメモ

tf.test.is_built_with_cuda()

成果率と検索に強いホームページ制作・運用改善のご相談を承っています。
法人クライアントとの契約多数、WEB制作会社やIT専門人材との協業、サイト保守・運用まで幅広く対応しています。

（2025年7月現在）

サイト改善・集客最適化まで、一貫してお任せいただけます。

当サイトはリンクフリーです

引用・SNSでの共有はご自由にどうぞ。事前連絡は不要ですが、もしご連絡いただければ、貴社コンテンツをご紹介できる場合もございます。