rocm/pytorch-xdit
rocm
基于ROCm的xDiT镜像，用于运行并行扩散推理
下载次数: 0状态：社区镜像维护者：rocm仓库类型：镜像最近更新：13 天前
让 AI 帮你使用轩辕镜像？ · 展开查看说明 · 点击收起说明
如果你使用 DeepSeek、元宝 AI、Copilot、ChatGPT 等 AI 工具，协助配置轩辕镜像、编写 docker pull 命令、修改 Docker Compose 镜像地址、配置镜像加速、排查镜像拉取失败、分析报错日志等问题，请先让 AI 阅读并遵守轩辕镜像的规则文档。
只需在 AI 对话中先发送下面这段话即可：
请先阅读并遵守：https://xuanyuan.cloud/agents.md
未读文档前不要生成 pull 命令或排错方案。查看 agents.md 用法指南与完整示范。国内用户首推 元宝 AI、DeepSeek 的深度思考模式，不推荐豆包 AI；Cursor 等编辑器可在对话 @ 该链接，或加入 User Rules。 若 AI 无法访问外链，可 打开说明文档 复制全文粘贴。文档会随站点更新，复制内容可能过期，建议定期检查。
PyTorch xDiT 镜像
镜像概述和主要用途
基于ROCm的镜像，用于通过https://github.com/xdit-project/xDiT运行并行扩散推理。该镜像专为在AMD ROCm生态系统中部署扩散模型推理任务设计，提供了高效的GPU加速支持。
核心功能和特性
- 基于ROCm平台构建，支持AMD GPU加速
- 集成xDiT项目，实现并行扩散推理能力
- 适用于PyTorch框架下的扩散模型推理场景
- 提供稳定版本及开发预览版本支持
使用场景和适用范围
适用于需要进行扩散模型并行推理的AI应用场景，尤其适合在AMD ROCm环境中部署的深度学习推理任务，可用于研究、开发及生产环境中的扩散模型推理工作负载。
使用方法和配置说明
详细使用指南
有关完整的使用说明、配置步骤和最佳实践，请参考官方文档：https://rocm.docs.amd.com/en/latest/how-to/rocm-for-ai/inference/xdit-diffusion-inference.html
镜像获取
稳定版本
可通过Docker Hub获取官方稳定版本镜像（具体标签请参考官方文档）
预览和开发版本
开发预览版镜像托管于Docker Hub：https://hub.docker.com/r/amdsiloai/pytorch-xdit，包含最新功能和更新，适合开发测试使用。
镜像拉取方式
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问 标签列表页面。
轩辕镜像加速拉取命令点我查看更多 pytorch-xdit 镜像标签
docker pull docker.xuanyuan.run/rocm/pytorch-xdit:<标签>
DockerHub 原生拉取命令
docker pull rocm/pytorch-xdit:<标签>
镜像拉取常见问题
功能
错误码
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"