huggingface/accelerate Docker 镜像 - 轩辕镜像

Hugging Face Accelerate 官方Docker镜像

Accelerate在CI流程中发布多种Docker版本供用户使用。这些是稳定的镜像，Accelerate可基于其运行，并包含多种预设配置，所有镜像均官方托管于Docker Hub。

以下是各镜像的详细说明。

命名约定

Accelerate Docker镜像遵循以下标签命名规则：

bash
huggingface/accelerate:{accelerator}-{nightly,release}

其中accelerator代表预配置的后端支持类型：

• gpu：基于nvidia/cuda镜像构建，包含完整依赖（如deepspeed、bitsandbytes等）
• cpu：基于python:3.8-slim镜像构建，适用于非CUDA workloads
• 更多类型即将推出

Nightlies vs Releases

Release版本

每个Release版本会推送包含版本号的镜像。例如，0.28.0版本的GPU支持镜像标签为：

bash
huggingface/accelerate:gpu-release-0.28.0

Nightly版本

Nightly镜像包含两种标签：

• nightly：每日构建的通用标签
• nightly-YYYY-MM-DD：特定日期的构建标签

例如，2024年3月14日的CPU nightly镜像标签为：

bash
huggingface/accelerate:cpu-nightly-2024-03-14

运行镜像

所有镜像均预装conda，并包含名为accelerate的环境（已安装所有依赖）。

拉取镜像

拉取最新nightly版本（以GPU为例）：

bash
docker pull huggingface/accelerate:gpu-nightly

运行镜像

以交互式模式运行GPU镜像（启用GPU内存访问）：

bash
docker container run --gpus all -it huggingface/accelerate:gpu-nightly

废弃镜像

历史CPU和GPU镜像曾托管于huggingface/accelerate-gpu和huggingface/accelerate-cpu。这些构建已过时且不再更新。

对应的huggingface/accelerate:{gpu,cpu}镜像使用相同的Dockerfile，只需将镜像标签更新为上述新标签即可。为兼容历史用途，这些废弃镜像不会被删除，但将不再接收更新。