目錄

• 寫在前面
• 目錄結構
• 完整代碼
• 運行過程
• 碰到的坑

寫在前面

? ? ? ?因為要用TensorRT寫自定義層，一定會用到cuda，所以學一下。

目錄結構

? ? ? ?– <my_dir> – fuc.cu
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? fuc.h
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? test.cu
? ? ? ?? ? ? ?? ? ? ?? ? ? ? ? CMakeLists.txt

完整代碼

• fuc.cu

#include <stdio.h> #include "fuc.h"__global__ void kernel() {printf("hello world!"); }

• fuc.h

extern __global__ void kernel();

• test.cu

#include <stdio.h> #include "fuc.h"int main(void) {kernel<<<1,1>>>();cudaDeviceSynchronize();printf("hello world!\n");return 0; }

• CMakeLists.txt

cmake_minimum_required(VERSION 2.8) project(helloworld C CXX)# find_package(<Name>)命令首先會在模塊路徑中尋找 Find<name>.cmake，這是查找庫的一個典型方式 # QUIET 可選字段，表示如果查找失敗，不會在屏幕進行輸出（但是如果指定了REQUIRED字段，則QUIET無效，仍然會輸出查找失敗提示語） find_package(CUDA QUIET REQUIRED) # 下面這行注釋掉的代碼和上面那行代碼起到一樣的作用 # include(/usr/share/cmake-3.5/Modules/FindCUDA.cmake)set(CUDA_INCLUDE_DIRS /usr/local/a-9.0/include) set(CUDA_LIBRARY_DIRS /usr/local/cuda-9.0/lib64)include_directories(${CUDA_INCLUDE_DIRS}) link_directories(${CUDA_LIBRARY_DIRS})cuda_add_executable(helloworld test.cu fuc.cu)

運行過程

• 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函數重復定義”的錯誤，我還沒懂是為什么。

cmake_minimum_required(VERSION 2.8) project(helloworld C CXX) find_package(CUDA QUIET REQUIRED)set(CUDA_INCLUDE_DIRS /usr/local/a-9.0/include) set(CUDA_LIBRARY_DIRS /usr/local/cuda-9.0/lib64)include_directories(${CUDA_INCLUDE_DIRS}) link_directories(${CUDA_LIBRARY_DIRS})# 原： # cuda_add_executable(helloworld test.cu fuc.cu)# 改為以下兩行： file(GLOB_RECURSE CURRENT_INCLUDE ./*.h ./*.hpp ./*.cuh) cuda_add_executable(cuda_test ${CURRENT_INCLUDE})

總結

以上是生活随笔為你收集整理的cuda学习笔记1 - hello world实战的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。