ubuntu - nvidia-smi 和 nvidia x 服务器设置之间的顺序不同

Question

当我运行命令 nvidia-smi ，我得到以下两个按总线 ID 排序的 GPU:

For GPU 0, 00000000:0A:00.0

For GPU 1, 00000000:41:00.0

但是，当我运行 NVIDIA X server Settings ，我可以获得以下按总线 ID 排序的 GPU:

For GPU 0, PCI:65:0:0

For GPU 1, PCI:10:0:0

因此，它们根据自己的方式来枚举 Bus-Id，具有不同的顺序。
有什么办法可以让订单一致吗？
操作系统:Ubuntu 16.04
主板:微星 X399(适用于 AMD 1950X)
(附注)
事实上，我的电脑在使用 dataparallel 时被卡住了。在多个 GPU 上进行深度学习的机制。我想不同的GPU订单是原因。

Answer 1

重要的顺序是 CUDA 顺序。

您可以enumerate CUDA 设备，看看哪个是哪个。

然后，您将能够使用 CUDA_VISIBLE_DEVICES 在未用于显示的 GPU 上运行程序, cudaSetDevice() 或 cuCtxCreate()