인프런 커뮤니티 질문&답변
작성한 질문수
텐서플로우 로컬 gpu 사용에 대해서 궁금합니다
작성
·
509
0
안녕하세요. 개인 gpu에서 tf2를 동작시키는데 메모리가 24기가인 그래픽카드입니다. 그런데 모델 훈련 때 gpu를 23기가 근처로 점유하는데, 이를 배치를 낮춰도 아래와 같이 점유하는데 또 따로 보통 설정을 해야 되나요? 아니면 기본적으로 이렇게 수행되는 것인가요? 감사합니다
답변 1
0
권 철민
지식공유자
안녕하십니까,
tensorflow는 일단 최초 기동시에 GPU의 가용 가능한 메모리를 자신의 전용 용도로 먼저 잡아 놓기 때문에 그렇게 보입니다.
그리고 TRAIN시에 자기가 잡은 메모리보다 더 많이 사용하면 GPU OOM 오류를 발생 시킵니다.
감사합니다.