c - 在 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
}