Windows11 + Anaconda + TF/Keras + GPU の簡単な設定方法覚え書き

 適切なgpuドライバーがインストールされていれば、コマンドからnvidia-smi

conda install tensorflow=2.10.0=gpu_py39h9bca9fa_0 (python 3.9)

conda install tensorflow=2.8.*=gpu_*
のようにビルドを指定すればcudatoolkitとcudnnが付属したtensorflowを入手できる
ビルドはconda searchで探せる

# gpu 確認
from tensorflow.python.client import device_lib
device_lib.list_local_devices()


この記事が気に入ったらサポートをしてみませんか?