欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

檢測(cè)tensorflow是否使用gpu進(jìn)行計(jì)算的方式

 更新時(shí)間:2020年02月03日 10:27:05   作者:匠人_C  
今天小編就為大家分享一篇檢測(cè)tensorflow是否使用gpu進(jìn)行計(jì)算的方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

如下所示:

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

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

其他方法:跑計(jì)算量大的代碼,通過(guò) nvidia-smi 命令查看gpu的內(nèi)存使用量。

以上這篇檢測(cè)tensorflow是否使用gpu進(jìn)行計(jì)算的方式就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

最新評(píng)論