GPU 메모리 초기화 방법 (feat. ubuntu)

1. 글을 쓰게 된 계기

 

모델을 학습시키다가 중지시키게 되면

GPU에 남은 메모리를 정리해야하는 경우가 생기게 된다.

 

2. GPU 메모리 초기화 방법

 

특정 프로세스 종료

sudo kill -9 <PID>
# sudo kill -9 1234

 

특정 GPU 메모리 전체 초기화

# 0번 GPU의 모든 프로세스를 종료합니다.
sudo fuser -v /dev/nvidia0
sudo fuser -v /dev/nvidia0 -k