import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
查看日誌信息若包含gpu信息,就是使用了gpu。
其他方法:跑計算量大的代碼,通過 nvidia-smi 命令查看gpu的內存使用量。
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
查看日誌信息若包含gpu信息,就是使用了gpu。
其他方法:跑計算量大的代碼,通過 nvidia-smi 命令查看gpu的內存使用量。
01 綜述 Opal 是愛奇藝大數據團隊研發的機器學習平臺,包含特徵生產、樣本構建、模型訓練、模型部署在內的多環節 Bigdata + AI 開發服務,內置多種訓練鏡像、