demo: 链接
动机
五一好不容易有点假期 ,想学下用C++进行cuda编程并用cmake编译,目标是实现主函数在cpu上跑,并调用另一个C++写的在gpu上跑的cuda程序,并且希望用cmake编译。
这方面资料网上还是能搜到一些,但内容把我看得比较迷乱,经常有bug。所以博主这里自己写了一个demo放在github上开源,内容很简单就是一个基本框架,方便大家往里面套代码。
demo
环境要求:
装了cuda的ubuntu系统,以及cmake。
程序介绍:
main.cc是在cpu上运行的普通C++代码,调用include文件夹里device.cu中的check_gpu函数。check_gpu函数起的作用就是调用cuda检测有几块gpu并输出显示。
要测试这个demo直接运行test.sh就行了,它会新建一个build文件夹在里面用cmake进行编译,并执行生成的执行文件test.o。输出结果为有几块可用gpu,比如有4块结果就是:
cuda:4