TensorflowがGPUまたはCPUで実行されているかどうかを確認します

すべてのブログすべてのモットー：強いことが唯一の選択肢になるまで、自分が本当にどれだけ強いかはわかりません。

0.はじめに

Tensorflow2がGPUまたはCPUの概要を使用しているかどうかを確認します

1.テキスト

注： GPUドライバーと、cudaやcudnnなどのツールが構成されている限り、プログラムの実行では、デフォルトで最初にGPUが使用され、実行できない場合はCPUが使用されます。

1.1テンソルフロー1.x

ログ情報を表示します。GPU情報が含まれている場合は、GPUを使用することを意味します

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

1.2テンソルフロー2.x

ログ情報を表示します。GPU情報が含まれている場合は、GPUを使用することを意味します

import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf
sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(log_device_placement=True))

参照

[1] https://blog.csdn.net/weixin_42702666/article/details/88067012
[2] https://zhuanlan.zhihu.com/p/110151684
[3] https://blog.csdn.net/qi_yue_yu /article/details/103549345?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.nonecase