如果我在我的系统中有多个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?