GPU 사용량 모니터링

사용량 모니터링

# gpu
watch -n 1 -d nvidia-smi
# cpu, ram
htop

---

특정 GPU를 사용하고 싶을 때

CUDA_VISIBLE_DEVICES=(사용하고 싶은 GPU number) <python_command_line>        

예시

python main_train.py   

위의 커맨드를 0번 GPU로 돌리고 싶을 때 아래처럼 입력

CUDA_VISIBLE_DEVICES=0 python main_train.py   

현재 GPU는 0, 1, 2번 존재