텐서플로우 로컬 gpu 사용에 대해서 궁금합니다

Question

안녕하세요. 개인 gpu에서 tf2를 동작시키는데 메모리가 24기가인 그래픽카드입니다. 그런데 모델 훈련 때 gpu를 23기가 근처로 점유하는데, 이를 배치를 낮춰도 아래와 같이 점유하는데 또 따로 보통 설정을 해야 되나요? 아니면 기본적으로 이렇게 수행되는 것인가요? 감사합니다

Answer

안녕하십니까,

tensorflow는 일단 최초 기동시에 GPU의 가용 가능한 메모리를 자신의 전용 용도로 먼저 잡아 놓기 때문에 그렇게 보입니다.

그리고 TRAIN시에 자기가 잡은 메모리보다 더 많이 사용하면 GPU OOM 오류를 발생 시킵니다.

감사합니다.

유영재

텐서플로우 로컬 gpu 사용에 대해서 궁금합니다

이 글과 비슷한 Q&A

그랩마켓 웹 화면 구현하기 코드

Encoder Decoder 부터 Simple Chatbot까지 이상답변

from sklearn.datasets import load_boston

CNN가속기 질문