cuda学习笔记1 - hello world实战
生活随笔
收集整理的這篇文章主要介紹了
cuda学习笔记1 - hello world实战
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
目錄
- 寫在前面
- 目錄結構
- 完整代碼
- 運行過程
- 碰到的坑
寫在前面
? ? ? ?因為要用TensorRT寫自定義層,一定會用到cuda,所以學一下。
目錄結構
? ? ? ?– <my_dir> – fuc.cu
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? fuc.h
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? test.cu
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? CMakeLists.txt
完整代碼
- fuc.cu
- fuc.h
- test.cu
- CMakeLists.txt
運行過程
- cd <my_dir>
- mkdir build && pushd build
- cmake ..
- make
- ./helloworld 得到如下圖輸出:
- popd
碰到的坑
- 不要把test.cu文件命名為test.cpp,因為cpp文件會使用c++編譯器來處理,我們在test.cpp中include了func.h,同時在func.h中聲明了__global__函數。
- 使用如下CMakeLists.txt在make的時候會報“main函數重復定義”的錯誤,我還沒懂是為什么。
總結
以上是生活随笔為你收集整理的cuda学习笔记1 - hello world实战的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: TensorRT学习笔记6 - IPlu
- 下一篇: TensorRT学习笔记7 - 保存与读