檢測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.返回:
相關文章
相關標籤/搜索