참고 링크: github.com/tensorflow/tensorflow/issues/41990
Error
CUDA runtime implicit initialization on GPU:0 failed.
- Tensorflow 사용 버전: 2.2.0
가장 베이직한 시도는 Tensorflow 공식 홈에 가서 gpu 지원 관련 문서에 본인이 해당하는 오류를 검색하고 가이드라인을 따라가면 된다. 그러나 가끔 그런 것도 아예 안 먹힐 때가 있다. 그때부터는 Tensorflow에서 제공하는 github이슈나 Stackoverflow를 참고하면 좋다. 구글링도 좋지만 이러한 방법을 먼저 시도해보는 것도 튜토리얼 읽는 스킬이 늘어난다. (하지만 .. 나도 모르게 ctrl + c, v 중...)
Solution
다행히 Github에 나와있어서 해결할 수 있었습니다. NVCC, GPU 메모리 할당 코드도 안 된다면, 아래와 같은 GPU 메모리 용량 할당하는 코드를 작업할때 진행해주면 됩니다.
import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'
리눅스 환경에서 진행 중이라면 ~/.bashrc 하여 TF_FORCE_GPU_ALLOW_GROWTH=true 라고 하단에 입력해주고 저장합니다. =true에서 띄어쓰기하면 오류가 나기 때문에 붙여써야합니다.