在 Linux 上安装加速器（如 NVIDIA 加速器或其他 GPU 加速器）涉及多个步骤，具体方法取决于你使用的加速器类型和 Linux 发布版。以下是一个通用的安装指南，适用于大多数情况

步骤 1：准备安装

1. 检查硬件兼容性：

   • 确保你的 GPU 支持当前版本的加速器驱动。
   • 查看硬件制造商的官方网站,获取最新的驱动版本。

2. 安装必要的依赖软件包：

   • NVIDIA 加速器：
     • 如果你使用 NVIDIA GPU，安装以下依赖项：

       sudo apt install nvidia-dkms libnvidia-gl nvidia-opencl-dev

     • 如果你的系统使用了 apt,请确保仓库包含最新的驱动。

3. 卸载现有驱动（如果有）：

   • 如果你已经安装了旧版本的驱动,可能需要先卸载它们。
   • 对于 NVIDIA：

     sudo apt purge nvidia*
     sudo apt autoremove

4. 添加软件源：

   • 如果你使用的是第三方加速器（如 NVIDIA 的）,可能需要添加对应的软件仓库。
   • 对于 NVIDIA，推荐使用 jetson 或 main 仓库。

5. 更新系统：

   • 确保系统是最新的：

     sudo apt update && sudo apt upgrade

步骤 2：安装加速器驱动

根据你使用的加速器类型,安装对应的驱动。

安装 NVIDIA 加速器驱动

1. 下载驱动：

   • 访问 NVIDIA 官方网站，下载适合你的 Linux 版本的驱动。
   • 下载 nvidia-<version>-linux-<arch>，<version> 是驱动版本，<arch> 是架构（如 x86_64）。

2. 安装驱动：

   • 复制驱动文件到合适的位置（如 /usr/local/nvidia）：

     sudo cp nvidia-<version>-linux-<arch>/nvidia*
     sudo mv nvidia* /usr/local/nvidia

   • 设置驱动自动加载：

     sudo chmod -R 755 /usr/local/nvidia/

   • 加载驱动：

     sudo modprobe nvidia

   • 验证驱动版本：

     dpkg -l | grep nvidia

安装 OpenCl 加速器

如果你使用 OpenCl 加速器（如 AMD GPU）,需要安装相应的驱动和工具包。

1. 安装必要的软件包：
   • 对于 OpenCl：

     sudo apt install libclc-dev ocl-icd-opencl-dev

2. 安装驱动：
   • 下载 AMD 的 OpenCl 驱动，安装它们：

     sudo dpkg -i amdgpu-<version>-linux-<arch>-<driver_version>.deb

3. 验证 OpenCl 是否可用：
   • 运行以下命令检查 OpenCl 是否安装：

     clinfo

   • 如果输出显示支持 OpenCl,则驱动已成功安装。

步骤 3：验证加速器

安装完成后,验证加速器是否正常工作。

1. 查看加速器状态：

   • 对于 NVIDIA：

     nvidia-smi

   • 对于 AMD：

     amdgpu-smi

2. 运行加速测试：

   • 使用 OpenCL 的测试工具：

     /usr/local/bin/ocl-icd-test

   • 或者运行 CUDA 应用程序，确保它们能够利用 GPU 加速。

步骤 4：安装开发工具（可选）

如果你需要开发或编译加速器相关的软件,安装相应的开发工具。

安装 CUDA

如果你使用 NVIDIA 加速器和 CUDA：

1. 下载 CUDA 安装包，安装：

   sudo ./cuda-<version>-linux-<arch>-<hostname>.run

2. 配置 PATH：

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

安装 AMD ROCm

如果你使用 AMD GPU 和 ROCm：

1. 下载并安装 ROCm：

   sudo ./rocm-<version>-linux-<arch>-<hostname>.sh

2. 配置环境变量：

   source /opt/rocm/profile

步骤 5：故障排除

如果在安装过程中遇到问题,可以按照以下步骤排查：

1. 检查文件权限：
   • 确保驱动文件和目录有正确的权限：

     sudo chmod -R 755 /usr/local/nvidia

2. 检查模块加载：
   • 确保驱动已正确加载：

     lsmod | grep nvidia

3. 重启系统：
   • 有时需要重启系统才能使新驱动生效：

     reboot