728x90
분명 예전에 잘만 돌아가던 모델이었는데
오랜만에 다시 돌려보니까 학습 시작도 전에 메모리가 부족하다느니, 알고리즘을 찾을 수 없다느니,
뭐가 문제인지 찾아보니 GPU 문제인 것 같더라
InternalError: Dst tensor is not initialized.
갤럭시북 이온2에 MX450 써먹어서 텐서플로우 구동 시켜보려고 이런 저런 시도 많이 해봤는데
애초에 딥러닝 학습시키기에 성능 미달인 GPU였다. 파이토치는 지원도 안하는 것 같고.. 뒤늦은 깨달음
그때 cuda나 cuDNN 설치하면서 자동으로 GPU연산을 수행하는데
CPU 이용하기 위해 찾아온 코드이다. 한 줄이면 해결되는 걸 몇 시간 머리 싸매고 진짜.
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'728x90
'데이터사이언스 > 인공지능' 카테고리의 다른 글
| [ChatGPT 효과적으로 사용하기] 자동 번역기 '프롬프트 지니'가 필요한 이유 | 설치 방법 및 활용법 (0) | 2023.04.09 |
|---|---|
| PTB-XL ECG Dataset Description (0) | 2022.05.14 |
| 로컬에서 텐서플로우 gpu 사용 가능한 환경 설정 (CUDA, cuDNN 호환성 확인하기, MX450) (0) | 2022.02.15 |
| 기계번역 관련 참고자료 (1) | 2022.02.15 |
| Van Gogh의 화풍으로 인물 초상화 만들기 (Style Transfer 실습) (1) | 2021.12.22 |