chaque blog chaque devise: vous ne savez jamais à quel point vous êtes vraiment fort avant d'être fort est le seul choix que vous avez。
0. Préface
Vérifiez si Tensorflow2 utilise le GPU ou le résumé du CPU
1. Texte
Remarque: tant que le pilote GPU et des outils tels que cuda et cudnn sont configurés, l'exécution du programme utilisera d'abord le GPU par défaut, puis utilisera le CPU s'il ne peut pas être exécuté.
1.1 tensorflow 1.x
Afficher les informations du journal, s'il contient des informations GPU, cela signifie utiliser GPU
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
1.2 tensorflow 2.x
Afficher les informations du journal, s'il contient des informations GPU, cela signifie utiliser 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))
les références
[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-BlogCommendnonFromMase-blog-BlogCommendnonFromMase-2