nvidia-smi与pytorch中GPU顺序不一致问题

nvidia-smi与pytorch中观察到的GPU顺不一致

原因

nvidia-smi中使用PCI_BUS_ID排序，torch中使用FASTEST_FIRST排序

解决方案

export CUDA_DEVICE_ORDER=PCI_BUS_ID