Keras/Tensorflow選擇GPU/CPU運行

首先，導入os，再按照PCI_BUS_ID順序，從0開始排列GPU，

import os

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

然后就可以選擇用哪一個或者那幾個GPU運行：

os.environ[”zCUDA_VISIBLE_DEVICES”] = "0"

用0號GPU，即'/gpu:0'運行；

os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1”

用0號和1號設備，即'/gpu:0'、'/gpu:1'運行；

os.environ[“CUDA_VISIBLE_DEVICES”] = “7”

用7號設備，即'/gpu:7'運行，但是'/gpu:7'根本不存在，於是就只能用CPU運行了。