Linux安装CUDA

• 输入以下命令，查看 GPU 支持的最高 CUDA 版本。笔者这里显示的是 11.6 ，这意味着，安装的 CUDA 版本必须 <= 11.6 。

  nvidia-smi
  

  

• 前往 Nvidia 的 CUDA 官网：CUDA Toolkit Archive | NVIDIA Developer ，笔者下载 CUDA 11.6.2 版本：

  

• 如下图选择：

  

• 选择好后，下方会显示两条 Linux 命令。在 Linux 终端输入这条命令：

  $ wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
  

• 这里笔者的服务器无法翻墙，直接在服务器上下载会超时。所以直接复制下载地址，先在 Windows 系统下下载好，再上传到 /root/ 目录下。

  https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
  

  

• 这里由于笔者没有 root 权限，无法上传到 /root/ 目录下。因此将此安装包上传至 /home/og/xxx 自己的工作目录下。输入以下命令：

  $ sudo sh cuda_11.6.2_510.47.03_linux.run
  

• 弹出来第一个选择框，笔者这台服务器已经安装过 Nvidia 的显卡驱动了，因此选择 “continute” ，会出现下面这个对话框，选择 “accept” 。

  

• 笔者服务器已经安装过 Nvidia 显卡驱动了，因此不需要再安装了。如果你尚未安装驱动，可以顺便一起安装了。摁一下空格取消 Driver 安装，直接选择 Install 安装：

  

• 安装成功：

  

  

• 接下来配置环境变量，直接在 Linux 命令行输入以下命令：

  sudo vim ~/.bashrc
  

  在文件末尾添加 (这里笔者假设你已经熟悉 Linux 下 Vim 文本编辑器的使用，不熟悉的强烈推荐去学习一下) ：

  export PATH=/usr/local/cuda-11.6/bin${PATH:+:${PATH}}
  export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  export CUDA_HOME=/usr/local/cuda-11.6
  

• 更新系统环境：

  source ~/.bashrc
  sudo ldconfig
  

• 验证 CUDA 是否安装成功，输入以下命令：

  nvcc -V
  

  输出下图即为成功安装：