什么是cuda?
统一计算设备架构(Compute Unified Device Architecture, CUDA),是由NVIDIA推出的通用并行计算架构。解决的是用更加廉价的设备资源,实现更高效的并行计算。
什么是cuda工具包?
CUDA 工具包是一种针对支持CUDA功能的GPU(图形处理器)的C语言开发环境。CUDA 开发环境包括: nvcc C语言编译器、适用于GPU(图形处理器)的CUDA FFT和BLAS库、分析器、CUDA运行时(CUDA runtime)驱动程序、CUDA编程手册。
CUDA 开发者软件开发包(SDK)提供了一些范例(附有源代码),以帮助使用者开始CUDA编程。
1. 查看显卡驱动版本,找对应cuda版本关系
【注意】:
cuda有两种API,分为运行API和驱动API,只要驱动API版本高于运行时API就没关系。nvidia-smi显示的是驱动API,nvcc -V显示的是运行时API。
2. cuda安装
(1)下载cuda版本,官网地址:
https://developer.nvidia.com/rdp/cudnn-archive
输入下载命令:
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
(2)安装方法
安装命令:
sudo sh cuda_10.2.89_440.33.01_linux.run
Do you accept the above EULA? (accept/decline/quit):
accept
选择安装选项,不选驱动,驱动已手动安装。
静候几分钟,安装完毕。
(3)设置cuda的环境变量
输入命令:sudo gedit ~/.bashrc
在后面输入以下内容:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64export PATH=$PATH:/usr/local/cuda-10.2/binexport CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
输入命令:source ~/.bashrc 使之生效;
输入命令:nvcc -V 验证是否安装成功
如果显示如下文字就说明安装成功了:
3. cudnn安装
(1)下载cudnn版本,官网地址:
https://developer.nvidia.com/cuda-toolkit-archive
选择deb包下载;
(2)安装cudnn
sudo dpkg -i cudnn-local-repo-ubuntu1804-8.3.1.22_1.0-1_amd64.deb
(3)验证是否安装成功
输入命令:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
《完》
点击上方小卡片关注我
上一篇
Ubuntu手动安装Nvidia显卡驱动的方法