Deep Learning & Machine Learning/강좌&예제 코드

Tensorflow 디버깅 정보 메시지 안보이게 하기

webnautes 2021. 9. 14. 22:32
반응형

Tensorflow 모듈을 사용할 경우 사용자가 출력하지 않은 추가 정보들이 많이 출력됩니다. 

특히 CUDA를 Tensorflow 모듈에서 사용하는 경우 더 많이 출력됩니다. 

 

2021. 9. 14 최초작성

 

 

다음 링크에서 소개하는 방법 중 하나를 적용하면 이 디버깅 정보 메시지를 안보이게 할 수 있습니다. 

https://stackoverflow.com/questions/35911252/disable-tensorflow-debugging-information 

 

 

예를 들어 다음처럼 tensorflow 모듈만 임포트했는데도 CUDA 관련 메시지가 추가로 출력됩니다. 

 

C:\Users\webnautes>python
Python 3.7.7 (tags/v3.7.7:d7c567b08f, Mar 10 2020, 10:41:24) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow
2021-09-14 22:28:38.813567: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library cudart64_110.dll
>>>

 

 

위 링크에서 소개하고 있는 해결 방법 중 하나인 다음 2줄을 추가한 후 tensorflow 모듈을 임포트해봅니다.

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 

 

 

C:\Users\webnautes>python
Python 3.7.7 (tags/v3.7.7:d7c567b08f, Mar 10 2020, 10:41:24) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import os
>>> os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
>>> import tensorflow as tf
>>>

 

 

이번엔 CUDA 관련 메시지가 출력되지 않습니다. 

 

 

 

 

반응형