cuda cmake C++ demo

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值