cuda - 如何检查 cuda 可用性（C++）？

Question

是否可以在调用所有 cudaMalloc... 命令之前检查是否存在任何 CUDA 设备？

我正在使用 C++，我只想在程序启动之前打印一条错误消息，以防用户不支持 cuda 技术。

编辑：如果我可以从 C# 检查它，它会更好。谢谢！

Answer 1

您可以使用cudaGetDeviceCount获取 cuda 设备的数量并使用 cuda 设备属性来检索您必要的计算能力。

cudaGetDeviceCount可以在此处找到相当旧版本的 API 文档。