tensorflow 2.3.1 이후 버전부터는 GPU가 자동으로 연동된다. 이 때문에 GPU를 사용하지 않는 연산에서도 GPU 메모리를 점유하여 OOM 에러가 나거나 학습이 느려지는 현상이 발생한다. 아래와 같은 문장을 모델 컴파일/ 학습/예측/ 평가 이전에 추가해주면 GPU를 사용하지 않고 CPU만을 활용하여 연산한다.
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
반응형
'인공지능 > ML, DL' 카테고리의 다른 글
| [NLP] Sentence-transformer를 활용한 문장 임베딩 (0) | 2023.04.28 |
|---|---|
| [ChatGPT] Python ChatGPT API 사용 방법 (한국어 예시) (0) | 2023.04.27 |
| [tensorflow] 모델 로드 시 컴파일 warning 해결 (No training configuration found in the save file, so the model was *not* compiled.) (0) | 2022.12.19 |
| [ML, DL] Baseline Accuracy란? (기준 모델, 기준 정확도) (0) | 2022.08.30 |
| [CNN] XAI 구현하기 - (1) CAM (0) | 2022.03.07 |
