텐서플로우 CPU 이용해서 학습하기 (GPU X)

분명 예전에 잘만 돌아가던 모델이었는데

 

오랜만에 다시 돌려보니까 학습 시작도 전에 메모리가 부족하다느니, 알고리즘을 찾을 수 없다느니,

 

뭐가 문제인지 찾아보니 GPU 문제인 것 같더라

 

InternalError: Dst tensor is not initialized.

 

갤럭시북 이온2에 MX450 써먹어서 텐서플로우 구동 시켜보려고 이런 저런 시도 많이 해봤는데

 

애초에 딥러닝 학습시키기에 성능 미달인 GPU였다. 파이토치는 지원도 안하는 것 같고.. 뒤늦은 깨달음

 

그때 cuda나 cuDNN 설치하면서 자동으로 GPU연산을 수행하는데

 

CPU 이용하기 위해 찾아온 코드이다. 한 줄이면 해결되는 걸 몇 시간 머리 싸매고 진짜.

 

os.environ['CUDA_VISIBLE_DEVICES'] = '-1'