nvidia-smi에 쌓여있는 python 프로세스 한번에 처리하기

너 쌓여있잖아

https://cryptosalamander.tistory.com/186

[Python / Linux] GPU 메모리 비우기, 프로세스 전부 종료하기

본 포스팅은 해당 블로그의 글을 200% 참고하였습니다.
그냥 내가 다시 구글링할 키워드 찾기가 어려울 것 같아서 코드 복붙합니다. Respect합니다.

우선 프로세스가 쌓여있는 GPU Number를 확인해준다.

주의) 다음 코드를 입력시 해당 GPU에 올라간 모든 python process가 종료됩니다.
반드시 두번 세번 확인하고 본인의 프로세스만 돌아가고 있는지 꼭 확인하고 코드를 돌리세요!

for i in $(lsof /dev/nvidia0 | grep python | awk '{print $2}' | sort -u); do kill -9 $i; done

/dev/nvidia0의 0을 프로세스가 쌓여있는 본인 환경의 GPU 번호로 수정하고 해당 명령어를 입력한다.