Linux安装CUDA

Ubuntu20.04系统安装CUDA

前言

笔者最近在做深度学习和感知方面的一个学习，需要用到CUDA 来加速，所以安装了CUDA ，在这里总结了一下。

一、CUDA 是什么

cuda是Compute Unified Device Architecture的缩写。中文叫统一计算架构。是为了让Nvidia GPU（可以通俗理解成显卡，但是与显卡有不同）可以完成通用计算任务的一种集成技术。我们通常可以使用的cuda框架有c，c++,fortran,python,java的，可以为数据大吞吐量的工作提供很好的加速功能。

二、步骤

1.先查看有没有安装CUDA 在终端输入命令

nvcc -V 
#或者
nvcc --version

2.若没有，查看有没有安装显卡驱动

Software&Updates --> Additional Drivers
#如下图

一般默认是底下那个

open source

，一般不用这个，改成一个带有

recommend

的，笔者这里是

470

3.查看Driver对应的CUDA version

终端输入

nvidia-smi

,显示

笔者是

CUDA 11.4

4.到cuda-toolki-archive,下载对应的cuda

笔者选择了

cuda 11.4.4

选择了对应的属性之后，在终端依次输入蓝色框命令，即：

wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda_11.4.4_470.82.01_linux.run
sudosh cuda_11.4.4_470.82.01_linux.run

这时候因为前面步骤已经安装了

Driver

，这里会提示

Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.

，我们选择continue即可，然后出现下面这样的界面，

enter

一下把

Driver

取消

安装过程结束后，显示下面界面

5.修改环境变量

vim ~/.bashrc

输入

exportPATH=/usr/local/cuda-11.4/bin${PATH:+:${PATH}}exportLD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}#其实这两句在步骤4安装完成后有说

接着

source ~/.bashrc

一下
最后再输入

nvcc -V

或者

nvcc --version

，显示

表示安装成功

总结

至此，系统下对应的CUDA 版本已经安装成功啦~