网络编程
位置:首页>> 网络编程>> Python编程>> 检测tensorflow是否使用gpu进行计算的方式

检测tensorflow是否使用gpu进行计算的方式

作者:匠人_C  发布时间:2021-11-19 11:24:33 

标签:tensorflow,gpu

如下所示:


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

查看日志信息若包含gpu信息,就是使用了gpu。

其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量。

来源:https://blog.csdn.net/castle_cc/article/details/78389082

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com