本期目录

  • Linux安装CUDA

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 ~/.bashrcsudo ldconfig

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

    nvcc -V

    输出下图即为成功安装：