CUDA_VISIBLE_DEVICE
-
cuda에서 gpu 할당dev 2018. 2. 2. 17:04
특정 gpu에만 gpu 할당하는 방법 Bash에서 실행 시 반영하려면 CUDA_VISIBLE_DEVICES=1 [command]CUDA_VISIBLE_DEVICES=1,2,3 [command] CUDA_VISIBLE_DEVICES 파이썬에서 코딩하기 출처https://stackoverflow.com/questions/37893755/tensorflow-set-cuda-visible-devices-within-jupyter/37901914import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VISIBLE_DEVICES"] = "0"