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

,下载对应的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
sudo sh 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

输入

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

接着source ~/.bashrc一下
最后再输入nvcc -V或者nvcc --version，显示

表示安装成功

总结

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

来源：C_oarnge