ubuntu18.04安装CUDA


文章目录

  • ubuntu18.04安装CUDA
  • 1.先要安装显卡驱动
  • 2. CUDA官网下载toolkit
  • 3. 安装CUDA
  • 4. 配置环境变量
  • 5. 查看CUDA版本


1.先要安装显卡驱动

可通过ubuntu Software & Updates中的Additional Drivers安装

比如可选择460或470驱动，笔者选择了460。此步骤不装也行，CUDA安装包中包含安装驱动。

容器内安装cuda驱动 cuda是否安装_容器内安装cuda驱动

点击Apply Changes可进行安装,安装完可输入下面命令查看是否安装成功

nvidia-smi # 查看是否安装成功

容器内安装cuda驱动 cuda是否安装_caffe_02

图中460已经安装成功，CUDA Version: 11.2表示可安装的CUDA最高版本为11.2，所以进入CUDA官网时，一定要选择该版本，否则会安装不成功。

2. CUDA官网下载toolkit

进入官网，找到CUDA Toolkit/SDK

容器内安装cuda驱动 cuda是否安装_深度学习_03

进入Toolkit下载页面，点击Download Now>

容器内安装cuda驱动 cuda是否安装_容器内安装cuda驱动_04

这部不要直接选择系统类型，选择下面的Archive of Previous CUDA Releases

容器内安装cuda驱动 cuda是否安装_深度学习_05

选择历史版本下载，笔者下载 CUDA Toolkit 11.2.0

容器内安装cuda驱动 cuda是否安装_深度学习_06

选择系统版本等信息，获取下载链接

容器内安装cuda驱动 cuda是否安装_CUDA_07

下载命令

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

cuda_11.2.0_460.27.04_linux.run 共3G，需要下载很久。

3. 安装CUDA

sudo sh cuda_11.2.0_460.27.04_linux.run

程序起来之后，会出现下面的提示，程序检测到电脑里面已经安装了驱动了，但是cuda还没安装，仍然继续，点击Continue

容器内安装cuda驱动 cuda是否安装_CUDA_08

会显示是否同意License的界面，输入accept

容器内安装cuda驱动 cuda是否安装_容器内安装cuda驱动_09

进入到CUDA 安装项选择界面，因为已经安装了Nvidia的驱动460，这一步跳过，不要勾选Driver，注意：X表示选择，空的中括号表示不选择。

容器内安装cuda驱动 cuda是否安装_CUDA_10

安装过程很快，2-3分钟。

4. 配置环境变量

打开 /home/username/.bashrc 文件，最后添加两行：

export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

5. 查看CUDA版本

安装好CUDA并配置好环境变量后，重启终端或另打开一个终端输入：

nvcc -V

容器内安装cuda驱动 cuda是否安装_pytorch_11


安装成功！