首先，匯入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執行了。