本期目录

  • 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

    输出下图即为成功安装: