目錄

• 寫在前面
• 目錄結(jié)構(gòu)
• 完整代碼
• 運(yùn)行過程
• 碰到的坑

寫在前面

? ? ? ?因?yàn)橐肨ensorRT寫自定義層，一定會(huì)用到cuda，所以學(xué)一下。

目錄結(jié)構(gòu)

? ? ? ?– <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>)命令首先會(huì)在模塊路徑中尋找 Find<name>.cmake，這是查找?guī)斓囊粋€(gè)典型方式 # QUIET 可選字段，表示如果查找失敗，不會(huì)在屏幕進(jìn)行輸出（但是如果指定了REQUIRED字段，則QUIET無效，仍然會(huì)輸出查找失敗提示語） 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)

運(yùn)行過程

• cd <my_dir>
• mkdir build && pushd build
• cmake ..
• make
• ./helloworld 得到如下圖輸出：
  
• popd

碰到的坑

• 不要把test.cu文件命名為test.cpp，因?yàn)閏pp文件會(huì)使用c++編譯器來處理，我們?cè)趖est.cpp中include了func.h，同時(shí)在func.h中聲明了__global__函數(shù)。
• 使用如下CMakeLists.txt在make的時(shí)候會(huì)報(bào)“main函數(shù)重復(fù)定義”的錯(cuò)誤，我還沒懂是為什么。

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})

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。