littlebot
Published on 2025-04-08 / 2 Visits
0

【源码】基于Linux的AI算力卡启动与驱动管理

项目简介

本项目是一个用于管理和启动AI算力卡的Linux驱动和启动流程工具集。支持多种硬件平台,像AX650、Intel、RK3588、Xilinx、NXP等,能在不同操作系统和内核版本上运行,具备算力卡检测、驱动加载、启动、虚拟网卡创建以及与算力卡通信等功能。

项目的主要特性和功能

  1. 多平台支持:支持多种硬件平台和操作系统,包括AX650、Intel、RK3588、Xilinx、NXP等,兼容Linux内核版本5.4至6.8。
  2. 算力卡检测:通过lspci命令检测算力卡是否存在,并验证设备ID和厂商ID。
  3. 驱动加载与编译:提供驱动编译和加载功能,支持不同平台的内核版本。
  4. 算力卡启动:通过axdl命令启动算力卡,并自动创建虚拟网卡。
  5. 虚拟网卡创建:为主板创建虚拟网卡,方便与算力卡进行通信。
  6. 通信测试:通过pingssh命令测试与算力卡的通信,确保算力卡正常运行。
  7. 错误处理与调试:提供错误处理和调试功能,确保驱动和算力卡启动过程中的问题能够被及时发现和解决。

安装使用步骤

假设用户已经下载了本项目的源码文件。

1. 检测算力卡

使用lspci命令检测算力卡是否被检测到: bash lspci 确保输出中包含厂商ID 1f4b和设备ID 0650

2. 加载驱动(适用于非AX650平台)

对于非AX650平台,需要先加载驱动。解压提供的驱动压缩包并编译: bash tar -zxvf x86_pcie.tgz cd x86_pcie make make install 加载驱动: bash sudo insmod ax_pcie_dev_host.ko 使用lsmod命令检查驱动是否加载成功。

3. 启动算力卡

解压axdl.tar.gz并启动算力卡: bash tar -zxvf axdl.tar.gz sudo ./axdl 等待30秒,算力卡将自动创建虚拟网卡,IP为192.168.1.1

4. 创建虚拟网卡

为主板创建虚拟网卡,IP为192.168.1.2bash sudo ./pcie_net2.sh master

5. 测试通信

使用ping命令测试与算力卡的通信: bash ping 192.168.1.1 通过ssh进入算力卡: bash ssh root@192.168.1.1 默认密码为123456

6. 卸载驱动(可选)

如果需要卸载驱动,可以使用以下命令: bash sudo rmmod ax_pcie_dev_host

注意事项

  • 不同平台和内核版本可能需要使用不同的驱动包,请根据实际情况选择合适的驱动包。
  • 在编译驱动时,确保系统已安装必要的编译工具和内核头文件。
  • 如果在编译过程中遇到Exec format error错误,请参考项目中的Troubleshooting部分进行解决。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】