여러 대의 GPU 중 선택하여 학습 진행하기 (코드 수정 없이)

 

Multi-GPU가 설치된 서버에서

여러 사람과 협업을 하게 되면 GPU를 나눠 쓰는 경우가 종종 있다.

 

여러 대의 GPU중 몇 개만 선택, 할당하여 학습에 사용하고 싶다면

 

먼저 할당 되어 있는 (다른사람이 사용하고 있는)

혹은 사용하고 싶은 GPU의 번호를 확인하고

 

$ nvidia-smi

 

 

사용할 GPU 번호를 결정하고 트레이닝 코드를 돌릴 시 아래와 같이 수행하자

(GPU는 0번부터 시작)

 

- GPU 2번만 사용하여 train.py 실행

$ CUDA_VISIBLE_DEVICES=2 python train.py

 

- GPU 0, 1번을 사용하여 train.py 실행

$ CUDA_VISIBLE_DEVICES=0,1 python train.py

 

위 방법으로 다른 사람의 코드를 코드 수정 없이 자신의 GPU를 할당하여 

학습을 진행할 수 있다.