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);
}
}