2016-03-08 62 views
2

任何人都知道是否有一種方法可以防止張量流與gpus的內存分配日誌污染標準錯誤? 我注意到,當執行以下命令:避免在標準錯誤上打印張量流

with tf.Session() as sess: 

在標準錯誤tensorflow打印日誌有關存儲器和GPU資源分配。例如:

I tensorflow/core/common_runtime/local_device.cc:25] Local device intra op parallelism threads: 48 
Graphics Device pciBusID 0000:02:00.0 
Free memory: 11.75GiB 
... 

出於重要原因,我想避免此打印。

+0

跟蹤問題:https://github.com/tensorflow/tensorflow/issues/1258 –

回答

4

這是最近修復的,如果升級到TensorFlow 0.12或更高版本,應該可用。

要禁用從TensorFlow日誌輸出都設置以下環境變量啓動的Python之前:

  • 0 =所有:

    $ export TF_CPP_MIN_LOG_LEVEL=3 
    $ python ... 
    

    還可以通過改變TF_CPP_MIN_LOG_LEVEL值調整的詳細程度消息被記錄(默認行爲)

  • 1 = INFO消息不打印
  • 2 = INFOWARNING消息不被印刷
  • 3 = INFOWARNING,並ERROR消息不被印刷
0

默認爲0,因此,所有的日誌被示出。將TF_CPP_MIN_LOG_LEVEL設置爲1以過濾出INFO日誌,2將其添加過濾掉WARNING,3以額外過濾掉ERROR。