- 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 - 소스코드에 직접 python code를 넣는 방법 :
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"참고 자료
'공부 > Tensorflow' 카테고리의 다른 글
| [Tensorflow] (B, C, W, H)를 (B, W, H, C)로 바꾸는 방법 (0) | 2019.08.27 |
|---|---|
| [Tensorflow] tensor scope/name 변경하기 (0) | 2019.08.27 |