CUDA安装

CUDA

Ubuntu 16.04 安装CUDA10

登录cuda官网,我用的是10.1版本,官网:CUDA Toolkit,按照自己的需求选择安装即可:

CUDA安装

安装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】:

CUDA安装

选择第一个,然后输入密码即可。

安装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

原文链接: https://www.cnblogs.com/muxuan/p/cuda_setup.html

欢迎关注

微信关注下方公众号,第一时间获取干货硬货;公众号内回复【pdf】免费获取数百本计算机经典书籍

    CUDA安装

原创文章受到原创版权保护。转载请注明出处:https://www.ccppcoding.com/archives/298077

非原创文章文中已经注明原地址,如有侵权,联系删除

关注公众号【高性能架构探索】,第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处!

(0)
上一篇 2023年2月15日 下午7:14
下一篇 2023年2月15日 下午7:15

相关推荐