CUDA安装

最新推荐文章于 2024-01-27 11:48:15 发布

weixin_30333885

最新推荐文章于 2024-01-27 11:48:15 发布

阅读量522

点赞数

文章标签： python c/c++

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

版权

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