查看是否用GPU跑的TensorFlow程序
第一種方法，直接輸出日志法（推薦）
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
第二種方法，跑計算量打的程序，然后用nvidia-smi查看GPU的使用情況
查看是否用GPU跑的TensorFlow程序
免責聲明！
本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。