[Tensorflow] GPU 할당 하는 방법

1. Terminal에서 python code를 실행하는 경우 :

$ CUDA_VISIBLE_DEVICES=0 python script.py  
$ CUDA_VISIBLE_DEVICES=1 python script.py  
$ CUDA_VISIBLE_DEVICES=2,3 python script.py  

2. 소스코드에 직접 python code를 넣는 방법 :

import os

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

참고 자료

• https://www.tensorflow.org/tutorials/using_gpu