Ubuntu20.04系统安装CUDA
- 前言
- 一、CUDA 是什么
- 二、步骤
- 1.先查看有没有安装CUDA 在终端输入命令
- 2.若没有，查看有没有安装显卡驱动
- 3.查看Driver对应的CUDA version
- 4.[到cuda-toolki-archive](https://developer.nvidia.com/cuda-toolkit-archive),下载对应的cuda
- 5.修改环境变量
- 总结
前言
笔者最近在做深度学习和感知方面的一个学习，需要用到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
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 版本已经安装成功啦~
