当我运行命令nvidia-smi
时,我会得到以下两个按总线ID排序的GPU:
对于GPU 00000000:0A:00.00
对于GPU 1,00000000:41:00.0
但是,当我运行NVIDIA X server Settings
时,我可以获得以下按总线ID排序的GPU:
对于GPU 0,PCI:65::0
对于GPU 1,PCI:10:0:0
因此,它们根据自己枚举总线Id的方式有不同的顺序。有没有办法使订单保持一致?
操作系统:Ubuntu 16.04主板:MSI X399(适用于AMD 1950X(
(PS(事实上,当在多个GPU上使用数据并行机制进行深度学习时,我的计算机被冻结了。我想不同的GPU顺序是原因。
重要的顺序是CUDA顺序。
您可以枚举CUDA设备,并查看哪一个是哪一个。
然后,您可以使用CUDA_VISIBLE_DEVICES
、cudaSetDevice()
或cuCtxCreate()
在未用于显示的GPU上运行程序
nvidia-smi
使用引导时向驱动程序注册GPU的顺序。CUDA首先使用最快的。