CUDA安裝


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

參考文獻

【1】https://wiki.ubuntu.com/UEFI/SecureBoot/DKMS


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM