确定计算机在运行时在CUDA C中是否具有GPU

Question

我正在启用GPU的计算机上编写代码，但是我的代码需要可移植到没有GPU的计算机上。所以我写了2个函数，一个仅使用CPU，另一个使用CPU + GPU。

我需要有条件的合规性代码，例如：

if (COMPUTER_HAS_GPU)
    //Run CPU+GPU code  
else  
    //Run CPU only code

有没有这样的东西？

Answer 1

您可以使用 cudaGetDeviceCount()

例如：

int devices = 0; 

cudaError_t err = cudaGetDeviceCount(&devices); 

if (devices > 0 && err == cudaSuccess) 
{ 
    // Run CPU+GPU code
} 
else
{ 
    // Run CPU only code
}