tlcpack/ci-gpu Docker 镜像 - 轩辕镜像

镜像概述

该公共Docker镜像提供支持CUDA GPU的持续集成（CI）环境，专门用于测试tlcpack和Apache TVM项目的构建过程。通过预配置的GPU加速环境，确保在CI流程中高效验证相关项目的构建结果。

核心功能与特性

• CUDA GPU支持：内置CUDA环境，可利用GPU资源加速测试流程
• CI流程适配：预配置CI所需依赖和工具，便于快速集成到自动化测试流程
• 项目针对性优化：针对tlcpack和Apache TVM的构建测试需求定制，确保兼容性
• 公共可访问：作为公共镜像提供，支持开发者和CI系统直接拉取使用

使用场景与适用范围

• tlcpack开发测试：验证代码提交或版本发布前的构建在GPU环境中的正确性
• Apache TVM构建验证：确保Apache TVM在CUDA GPU支持下的构建完整性
• GPU加速CI任务：适用于需要在GPU环境中执行的持续集成测试，尤其针对深度学习框架或高性能计算项目

使用方法

镜像拉取

从Docker仓库拉取镜像（具体名称及标签请参考GitHub仓库文档）：

bash
docker pull [镜像名称]:[标签]

CI集成示例（GitHub Actions）

在CI配置文件中使用该镜像：

yaml
jobs:
  gpu-ci-test:
    runs-on: [self-hosted, linux, gpu]
    container:
      image: [镜像名称]:[标签]
    steps:
      - name: Checkout code
        uses: actions/checkout@v4
      - name: Execute build tests
        run: |
          # 执行tlcpack/Apache TVM构建测试命令
          ./build.sh && ./run_tests.sh

参考资源

• GitHub仓库：[***]
• 官方网站：[***]