[우당탕탕] Ubuntu 서버에서 GPU 사용하기

python 실행 코드 내에서는

import os 
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"

python 실행 코드

CUDA_VISIBLE_DEIVCES=0 python run_code.py

GPU 번호는 0 말고 다른걸로 지정할 수 있다.

GPU 사용 여부 확인하기

nvidia-smi # 현시점에서의 gpu 사용량 확인 가능

또는

watch -n .1 -d nvidia-smi #0.1초간격으로 gpu사용량을 계속 확인가능