megatron_rl0.17.1-hygon-dtk26.04

前置条件 link

• 架构: x86_64
• 芯片型号: Hygon BW1000
• 宿主机驱动: 6.3.30-V1.4.1a
• 容器工具包 (可选) ⓘ: dcu-container-toolkit >= 1.3.0

镜像内容 link

基于 link

Python link

3.10

应用软件包 link

megatron-core[rl]==0.17.1

启动 link

已发布: harbor.baai.ac.cn/flagos-app/megatron_rl0.17.1-hygon-dtk26.04:2.2.0-0.2.3

镜像名较长——先将其设为变量：

  IMG=harbor.baai.ac.cn/flagos-app/megatron_rl0.17.1-hygon-dtk26.04:2.2.0-0.2.3
  

以下两种方式任选其一：

使用容器工具包 link

启动交互式 shell：

  docker run --rm -it \
  -e DCU_VISIBLE_DEVICES=all \
  $IMG bash
  

暂无启动器。 该镜像尚未提供启动器或默认命令——可先启动交互式 shell 查看镜像内容；启动器将随应用的入口点一并提供。

无需工具包——直接使用 docker / podman link

启动交互式 shell：

  docker run --rm -it \
  --device /dev/kfd \
  --device /dev/mkfd \
  --device /dev/dri \
  --group-add video \
  -v /opt/hyhal:/opt/hyhal \
  --security-opt seccomp=unconfined \
  $IMG bash
  

暂无启动器。 该镜像尚未提供启动器或默认命令——可先启动交互式 shell 查看镜像内容；启动器将随应用的入口点一并提供。