- tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
- 에러 메세지
문제 원인
처음에는 CUDA 버전 에러인 줄 알았다. (2080TI에 CUDA 10 사용 중)
그런데 저 에러가 뜨는 코드가 있고 안 뜨는 코드가 있었다.
그래서 cuda 버전 에러가 아닐 수도 있겠다고 생각했고 찾아본 결과 GPU 메모리 에러였다.
GPU 메모리를 초과해서 사용하는 경우 이런 에러가 발생할 수 있다고 했다.
해결 방법 : 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.75session = tf.Session(config=config)sessionset_env()