본문 바로가기

AI/Technique

Keras, Tensorflow에서 VRAM Control

1부

https://datamasters.co.kr/33

 

 

Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 1부

이번 포스팅에서는 Keras와 Tensorflow에서 GPU를 더 똑똑하게 사용하는 방법에 대해 알아보자. 케라스 (와 당연히 텐서플로우)를 사용한다면, GPU도 높은 확률로 사용 중일 것 이다. 근데 이놈의 텐서플로우는 d..

datamasters.co.kr

2부

https://datamasters.co.kr/48

 

Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 2부

혹시 잊었다면: Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 1부 1부에서는, tensorflow 를 통해 gpu와 그 메모리들을 관리하는 방법에 대해 다루었다. 이번 포스팅에서는 실제로 어떠한 방법들을 사용해야 g..

datamasters.co.kr

실제로 프로젝트를 진행하면서 OOM을 겪은 적이 자주 있었는데 잘 설명한 글이 있어서 링크를 첨부한다!