nvidia-smi 등으로 GPU 사용 현황을 확인할 때 현재 실행시키는 코드의 프로세스 이름을 원하는대로 등록해주기
-
실행하고자 하는 스크립트에서
setproctitle임포트from setproctitle import setproctitle
-
스크립트의 상단에 setproctitle('프로세스이름') 추가
setproctitle('lm_pretraining')
-
nvidia-smi로 프로세스 확인 시
Process name란에 lm_pretraining이라는 이름으로 뜨게 됨