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

こちらに詳しく解説されています。

TensorFlow 2.10 （GPU 対応可能）と Python のインストール，動作確認（Windows 上）

備忘録
tensorflowインストールから確認
テスト済みのビルド構成をチェック
CUDAでTensorFlowが実行できるか確認

備忘録

以下は、確か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'
    
}

tensorflowインストールから確認

pip install tensorflow

pipを使用してTensorFlowをインストールします

テスト済みのビルド構成をチェック

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

Build from source on Windows | TensorFlow

※備忘と参考程度のメモ

CUDAでTensorFlowが実行できるか確認

tf.test.is_built_with_cuda()

MKPのムジン社長

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