CUDA
Ubuntu 16.04 安裝CUDA10
登錄cuda官網,我用的是10.1版本,官網:CUDA Toolkit,按照自己的需求選擇安裝即可:
安裝CUDA
安裝CUDA按照CUDA官網的步驟即可,注意千萬不要隨意給電腦安裝什么驅動
,因為CUDA自帶的有驅動,也是最合適的驅動。
sudo dpkg -i cuda-repo-ubuntu1604-10-1-local-10.1.168-418.67_1.0-1_amd64.deb`
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
sudo apt-get update`
sudo apt-get install cuda`
有幾個問題
對於有些電腦有驅動保護,在你安裝CUDA過程中,會讓你輸入一個切換顯卡驅動的保護密碼,當你重啟電腦就會發現這個這個界面【1】:
選擇第一個,然后輸入密碼即可。
安裝CLion
下載地址:http://www.jetbrains.com/clion/
激活方式:https://help.lucoder.com/plugins/jetbrain/index.html
Hello word
CUDA的CmakeLists模板:
# 設置CMake版本
cmake_minimum_required(VERSION 3.5)
project(Test1)
set(EXECUTABLE_OUTPUT_PATH ../)
# 找到CUDA路徑並導入頭文件
find_package(CUDA REQUIRED)
include_directories(${CUDA_INCLUDE_DIRS})
# 設置C++版本
set(CMAKE_CXX_STANDARD 11)
# 檢索出源文件與頭文件
file(GLOB_RECURSE TEST_HEAD ${CMAKE_CURRENT_LIST_DIR}/src/.h ${CMAKE_CURRENT_LIST_DIR}/src/.cuh)
file(GLOB TEST_SRC ${CMAKE_CURRENT_LIST_DIR}/src/*.cpp ${CMAKE_CURRENT_LIST_DIR}/src/*.cu)
# 設置CUAD編譯配置
set(CUDA_NVCC_FLAGS "-g -G")
# build option
set(GENCODE -gencode=arch=compute_35,code=sm_35)
set(GENCODE ${GENCODE} -gencode=arch=compute_30,code=sm_30)
set(GENCODE ${GENCODE} -gencode=arch=compute_20,code=sm_20)
set(GENCODE ${GENCODE} -gencode=arch=compute_10,code=sm_10)
# 生成可執行文件
cuda_add_executable(Test ${TEST_SRC} ${TEST_HEAD} OPTIONS ${GENCODE})