koyo922
10/18/2017 - 1:49 AM

tensorflow utils

tensorflow utils

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())