nvidia-smi -L显示的GPU序号和代码中不同

1 GPU序号显示不同

• nvidia-smi -L 命令回显
  
• 代码中获取到的卡号
  

2 使用以下命令设置

说明： 码默认情况下设备排序是 FASTEST_FIRST
操作： 可直接在ssh终端执行一下命令，或直接保存到.bashrc环境变量中。

export CUDA_DEVICE_ORDER="PCI_BUS_ID" 

• 设置后效果如下：
  nvidia-smi -L 命令回显
  
  代码中获取到的卡号
  

3 获取GPU信息代码

    int devNum = 0;
    std::vector<cudaDeviceProp> m_vec_device_info;
    cudaGetDeviceCount(&devNum);
    for (int i = 0; i < devNum; i++)
    {
        cudaDeviceProp prop;
		if (cudaGetDeviceProperties(&prop, i) == cudaSuccess) 
		{
			m_vec_device_info.push_back(prop);
            ST_LOG_INFO("card id:{},message:{}",i,prop.name);
		}
    }