[ Tensorflow ] tensorflow load error

* Error Message

tensorflow.python.framework.errors_impl.UnknownError

 

* Problem
처음에는 CUDA 버전 에러인 줄 알았다. (2080TI에 CUDA 10 사용 중)
그런데 저 에러가 뜨는 코드가 있고 안 뜨는 코드가 있었다.
그래서 cuda 버전 에러가 아닐 수도 있겠다고 생각했고 찾아본 결과 GPU 메모리 에러였다.
GPU 메모리를 초과해서 사용하는 경우 이런 에러가 발생할 수 있다고 했다.

 

* Solution : gpu_memory를 적당양 사용하니까 문제가 해결됬다. ( 여기서는 75% 사용 )

# Tensor 환경 설정 
def set_env(): 
    os.environ['CUDA_VISIBLE_DEVICES'] = '0' 
    config = tf.ConfigProto() 
    config.gpu_options.per_process_gpu_memory_fraction = 0.75 
    session = tf.Session(config=config) 
    session 
set_env()