GPU 장치 번호가 일치하지 않습니다.

Question

Caffe에 GPU를 사용하고 있습니다. 내가 nvidia-smi을 할 때,

| NVIDIA-SMI 352.63  Driver Version: 352.63   |      
|-------------------------------+----------------------+----------------------+ 
| GPU Name  Persistence-M| Bus-Id  Disp.A | Volatile Uncorr. ECC | 
| Fan Temp Perf Pwr:Usage/Cap|   Memory-Usage | GPU-Util Compute M. | 
|===============================+======================+======================| 
| 0 GeForce GTX 750 Ti Off | 0000:01:00.0  On |     N/A | 
| 44% 52C P0  2W/38W | 464MiB/2047MiB |  0%  Default | 
+-------------------------------+----------------------+----------------------+ 
| 1 Quadro M4000  Off | 0000:02:00.0  Off |     N/A | 
| 59% 74C P0 66W/120W | 7434MiB/8191MiB |  95%  Default | 


+-----------------------------------------------------------------------------+ 
| Processes:              GPU Memory | 
| GPU  PID Type Process name        Usage  | 
|=============================================================================| 
| 0  1428 G /usr/lib/xorg/Xorg        316MiB | 
| 0  2200 G compiz           139MiB | 
| 1  29863 C ./caffe-segnet/build/tools/caffe    7413MiB | 
+-----------------------------------------------------------------------------+ 

을 보여줍니다하지만 CAFFE을 실행하고 GPU 하나를 선택할 때, 그것은 메모리가 나에게 말할 것이다. GPU 0 build/tools/caffe train -gpu 0 -solver solver.prototxt을 선택하면 실행할 수 있습니다.

왜 그렇습니까?

Answer 1

Nvidia CUDA는 어떤 장치가 가장 빠른 것으로 간주되는지에 따라 고유 한 장치 번호를 사용합니다. 그러나 PCI 버스 ID로 장치를 구할 수도 있습니다.

cudaError_t cudaDeviceGetByPCIBusId (int* device, char* pciBusId) 

또한 다른 방식으로 작동하며 특정 장치에 대한 버스 ID를 얻을 수 있습니다.

cudaError_t cudaDeviceGetPCIBusId (char* pciBusId, int len, int device) 

버스 ID는 값 0000:01:00.0 및 테이블에서 0000:02:00.0 될 것이다. 형식은 [domain]:[bus]:[device].[function]입니다.

[1] 및 [2]을 참조하십시오.