[Python] GPU 지정하여 Python 코드 실행하기

 

 

가끔 GPU를 나눠 할당하여 코드를 실행하고 싶을 때가 있는데 아래와 같이 간단히 설정 가능하다.

애용하는 방법이다. 

 

 

 

Terminal에서 python 코드를 실행하는 경우

~$ CUDA_VISIBLE_DEVICES=0 python test.py
~$ CUDA_VISIBLE_DEVICES=1 python test.py
~$ CUDA_VISIBLE_DEVICES=2,3 python test.py 

 

python 코드에서 GPU 지정하는 방법

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

 

 

 

참고자료 : dongjinlee.tistory.com/entry/%EC%84%A0%ED%83%9D%ED%95%9C-GPU%EC%97%90%EB%A7%8C-%EB%A9%94%EB%AA%A8%EB%A6%AC-%ED%95%A0%EB%8B%B9%ED%95%98%EB%8A%94-%EB%B0%A9%EB%B2%95

선택한 GPU에만 메모리 할당하는 방법