【03】如何檢驗 tensorflow-gpu 安裝成功 ?win10+CUDA9.0+cuDNN7.0+python3.6+tensorflow_gpu1.10 構建可用環境

參考:python https://blog.csdn.net/castle_cc/article/details/78389082c++ 編寫 demo.py import tensorflow as tf sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 查看日誌信息若包含gpu信息,就是使用了gpu。git
相關文章
相關標籤/搜索