更新时间:2022-07-29 GMT+08:00
分享

安装并配置GPU驱动

背景信息

对于使用GPU的边缘节点,在纳管边缘节点前,需要安装并配置GPU驱动。

IEF当前支持Nvidia Tesla系列P4、P40、T4等型号GPU,支持CUDA Toolkit 8.0至10.0版本对应的驱动。

操作步骤

  1. 安装GPU驱动。

    1. 下载GPU驱动,推荐驱动链接:

      https://www.nvidia.com/content/DriverDownload-March2009/confirmation.php?url=/tesla/440.33.01/NVIDIA-Linux-x86_64-440.33.01.run&lang=us&type=Tesla

    2. 执行如下安装驱动命令。

      bash NVIDIA-Linux-x86_64-440.33.01.run

    3. 执行如下命令检查GPU驱动安装状态。

      nvidia-smi

  2. 以root用户登录边缘节点。
  3. 执行如下命令。

    nvidia-modprobe -c0 -u

  4. 创建文件夹。

    mkdir -p /var/IEF/nvidia/drivers /var/IEF/nvidia/bin /var/IEF/nvidia/lib64

  5. 拷贝驱动文件。

    • 对于CentOS,依次执行如下命令拷贝驱动文件:

      cp /lib/modules/{当前环境内核版本号}/kernel/drivers/video/nvi* /var/IEF/nvidia/drivers/

      cp /usr/bin/nvidia-* /var/IEF/nvidia/bin/

      cp -rd /usr/lib64/libcuda* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/libEG* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/libGL* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/libnv* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/libOpen* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/libvdpau_nvidia* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib64/vdpau /var/IEF/nvidia/lib64/

    • 对于Ubuntu,依次执行如下命令拷贝驱动文件:

      cp /lib/modules/{当前环境内核版本号}/kernel/drivers/video/nvi* /var/IEF/nvidia/drivers/

      cp /usr/bin/nvidia-* /var/IEF/nvidia/bin/

      cp -rd /usr/lib/x86_64-linux-gnu/libcuda* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/libEG* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/libGL* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/libnv* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/libOpen* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/libvdpau_nvidia* /var/IEF/nvidia/lib64/

      cp -rd /usr/lib/x86_64-linux-gnu/vdpau /var/IEF/nvidia/lib64/

    其中,当前环境内核版本号可以使用uname -r命令查看获取,如下所示,请替换为实际取值。

    # uname -r
    3.10.0-514.e17.x86_64

  6. 执行以下命令修改目录权限。

    chmod -R 755 /var/IEF

相关文档