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

Question

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

我正在使用C ++，我只想在程序启动之前打印错误消息，因为用户不支持cuda tech。

编辑：如果我可以从C＃中查看它，它会更好。 谢谢！

Answer 1

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

可以找到cudaGetDeviceCount的{​​{1}}的旧版API文档。