import os os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'#对显卡进行编号(从0开始) os.environ['CUDA_VISIBLE_DEVICES'] = '1'#使用编号为1的显卡