如果我在我的系統中有多個Nvidia GPU，如何檢查CUDA編譯器當前使用的GPU？

Question

我有一個Windows系統與2個Nvidia GPUs。有人能告訴我哪個GPU是CUDA編譯器使用的嗎？是否可以切換GPU或將兩者一起使用以實現相同的過程？

Answer 1

http://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__DEVICE.html#group__CUDART__DEVICE_1g1bf9d625a931d657e08db2b4391170f0

使用 'cudaGetDeviceCount' 可能的複製來獲得設備的數量。如果deviceCount爲2，則設備索引0和設備索引1指的是兩個當前設備。

'cudaGetDeviceProperties'可用於獲取設備的許多屬性。

例如，

cudaDeviceProp deviceProp; 
cudaGetDeviceProperties(&deviceProp, 1); 

可用於獲得裝置1

而且方式的許多特性可切換到不同的GPU是容易的。初始化完成後，使用

'cudaSetDevice(0)' 

和

'cudaSetDevice(1)' 

切換到不同的GPU。

Answer 2

CUDA_VISIBLE_DEVICES環境變量將允許您修改此啓用/排序。

CUDA_VISIBLE_DEVICES="0,1"將使兩個GPU設備可用於您的程序。

的CUDA GPU selected by position, but how to set default to be something other than device 0?