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

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

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

備忘録

以下は、確か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本体のバージョンとの互換性をチェックします。

https://www.tensorflow.org/install/source_windows#tested_build_configurations

※備忘と参考程度のメモ

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

tf.test.is_built_with_cuda()