如果機器中有多塊GPU，tensorflow會預設吃掉所有能用的視訊記憶體， 如果實驗室多人公用一臺伺服器，希望指定使用特定某塊GPU。

可以在檔案開頭加入如下程式碼：

import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 照PCI_BUS_ID順序從0開始排列GPU裝置 os.environ["CUDA_VISIBLE_DEVICES"] = "1"       # 使用第二塊GPU（從0開始）

也可以制定使用某幾塊GPU

import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0, 2"    # 使用第一, 三塊GPU

禁用GPU

import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

如果寫

os.environ["CUDA_DEVICE_ORDER"] = "1" # 如果只有一塊gpu，將會用cpu