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運行了。