[환경] 딥러닝 RAM 부족, 메모리 부족

메모리 확인: nvidia-smi

 

 

방법 1. 배치를 나눈다.

 

 

방법 2. 모든 업무를 gpu로 하지 않을수 있다.

기타 잡 업무는 cpu로 처리를 하고 나머지 중요한 연산을 gpu로 진행하도록 조절한다.(코드로. with tf.device ??)

import os 
# GPU를 아예 못 보게 하려면: 
os.environ["CUDA_VISIBLE_DEVICES"]='' 
# GPU 0만 보게 하려면: 
os.environ["CUDA_VISIBLE_DEVICES"]='0' 
# GPU 1만 보게 하려면: 
os.environ["CUDA_VISIBLE_DEVICES"]='0' 
# GPU 0과 1을 보게 하려면: 
os.environ["CUDA_VISIBLE_DEVICES"]='0,1'

#출처: https://datamasters.co.kr/33 [데이터마스터]