Ubuntu20.04系统安装CUDA

  • 前言
  • 一、CUDA 是什么
  • 二、步骤
    • 1.先查看有没有安装CUDA 在终端输入命令
    • 2.若没有,查看有没有安装显卡驱动
    • 3.查看Driver对应的CUDA version
    • 4.[到cuda-toolki-archive](https://developer.nvidia/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/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 版本已经安装成功啦~

更多推荐

Linux安装CUDA