tensorflow(GPU)使用

一，直接指定GPU:html tf.ConfigProto通常用在建立session的時候。用來對session進行參數配置session with tf.Session(config = tf.ConfigProto(...),...)   #tf.ConfigProto()的參數 log_device_placement=True : 是否打印設備分配日誌 allow_soft_placeme

>>阅读原文<<