多GPU服务器指定GPU

在使用服务器的时候在终端指定了第3块GPU，但代码还是默认使用第1块GPU,在代码中加入下列几行代码即可：

import os

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1, 2"    # 使用第2, 3块GPU