检测tensorflow是否使用GPU(tensorflow-gpu版本怎么安装匹配的cuda和cudnn)

1.前提:本机已安装tensorflow-gpu 2.检测: import tensorflow as tf sess=tf.Session(config=tf.ConfigProto(log_device_placement=True)) 3.目的: 查看日志信息,若包含gpu信息,就是使用了gpu。 其他方法:跑计算量大的代码,通过 nvidia-smi 命令查看gpu的内存使用量 4.返回:
相关文章
相关标签/搜索