cada blog, cada lema: Você nunca sabe o quão forte você realmente é, até que ser forte seja a única escolha que você tem。
0. Prefácio
Verifique se o Tensorflow2 usa GPU ou resumo da CPU
1. Texto
Nota: Contanto que o driver da GPU e ferramentas como cuda e cudnn estejam configurados, a execução do programa usará a GPU primeiro por padrão e, em seguida, usará a CPU se não puder ser executada.
1,1 tensorflow 1.x
Ver informações de registro, se contiver informações de GPU, significa usar GPU
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
1,2 tensorflow 2.x
Ver informações de registro, se contiver informações de GPU, significa usar 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))
referências
[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-BlogCommendFromaiFromMachine2.nonecase