热门搜索:
rocm/pytorch-xdit
rocm
基于ROCm的xDiT镜像，用于运行并行扩散推理
下载次数: 0状态：社区镜像维护者：rocm仓库类型：镜像最近更新：1 个月前
让 AI 帮你使用轩辕镜像？ · 展开查看说明
如果你使用 DeepSeek、元宝 AI、Copilot、ChatGPT 等 AI 工具，协助配置轩辕镜像、编写 docker pull 命令、修改 Docker Compose 镜像地址、配置镜像加速、排查镜像拉取失败、分析报错日志等问题，请先让 AI 阅读并遵守轩辕镜像的规则文档。
只需在 AI 对话中先发送下面这句话即可：
请先完整阅读并严格遵守以下文档中的全部规则与要求：
https://xuanyuan.cloud/agents.md
在未充分阅读并理解该文档前，不要生成任何命令、配置、修改建议、故障排查方案或技术回答。后续所有输出都必须严格以该文档中的规范为最高优先级执行。查看 agents.md 用法指南与完整示范。国内用户首推 元宝 AI、DeepSeek 的深度思考模式，不推荐豆包 AI；Cursor 等编辑器可在对话 @ 该链接，或加入 User Rules。 若 AI 无法访问外链，可 打开说明文档 复制全文粘贴。文档会随站点更新，复制内容可能过期，建议定期检查。
PyTorch xDiT 镜像
镜像概述和主要用途
基于ROCm的镜像，用于通过https://github.com/xdit-project/xDiT%E8%BF%90%E8%A1%8C%E5%B9%B6%E8%A1%8C%E6%89%A9%E6%95%A3%E6%8E%A8%E7%90%86%E3%80%82%E8%AF%A5%E9%95%9C%E5%83%8F%E4%B8%93%E4%B8%BA%E5%9C%A8AMD ROCm生态系统中部署扩散模型推理任务设计，提供了高效的GPU加速支持。
核心功能和特性
- 基于ROCm平台构建，支持AMD GPU加速
- 集成xDiT项目，实现并行扩散推理能力
- 适用于PyTorch框架下的扩散模型推理场景
- 提供稳定版本及开发预览版本支持
使用场景和适用范围
适用于需要进行扩散模型并行推理的AI应用场景，尤其适合在AMD ROCm环境中部署的深度学习推理任务，可用于研究、开发及生产环境中的扩散模型推理工作负载。
使用方法和配置说明
详细使用指南
有关完整的使用说明、配置步骤和最佳实践，请参考官方文档：rocm-xdit-diffusion-inference
镜像获取
稳定版本
可通过Docker Hub获取官方稳定版本镜像（具体标签请参考官方文档）
预览和开发版本
开发预览版镜像托管于Docker Hub：https://hub.docker.com/r/amdsiloai/pytorch-xdit%EF%BC%8C%E5%8C%85%E5%90%AB%E6%9C%80%E6%96%B0%E5%8A%9F%E8%83%BD%E5%92%8C%E6%9B%B4%E6%96%B0%EF%BC%8C%E9%80%82%E5%90%88%E5%BC%80%E5%8F%91%E6%B5%8B%E8%AF%95%E4%BD%BF%E7%94%A8%E3%80%82
镜像拉取方式
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问 标签列表页面。
轩辕镜像加速拉取命令点我查看更多 pytorch-xdit 镜像标签
docker pull docker.xuanyuan.run/rocm/pytorch-xdit:<标签>
DockerHub 原生拉取命令
docker pull rocm/pytorch-xdit:<标签>
更多 pytorch-xdit 镜像推荐
rocm/pytorch
rocm
基于ROCm的PyTorch Docker镜像为开发者提供了预配置的深度学习环境，集成了PyTorch框架与AMD ROCm开源计算平台，支持AMD GPU硬件加速，无需手动配置驱动及库依赖，可直接用于深度学习模型的开发、训练与部署，有效简化环境搭建流程，确保跨平台一致性，适用于科研机构、工业界等多种场景，助力高效利用AMD硬件资源开展AI相关任务。
123 次收藏100万+ 次下载
12 天前更新
bitnamicharts/pytorch
bitnamicharts
Bitnami提供的PyTorch Helm chart，用于在Kubernetes环境中简化PyTorch的部署与管理。
50万+ 次下载
4 天前更新
semitechnologies/img2vec-pytorch
semitechnologies
暂无描述
10万+ 次下载
3 年前更新
rocm/pytorch-nightly
rocm
该Docker仓库用于托管PyTorch每日构建版Docker镜像。
7 次收藏5万+ 次下载
1 个月前更新
intel/intel-optimized-pytorch
intel
用于在英特尔架构上运行PyTorch工作负载的容器
16 次收藏10万+ 次下载
1 个月前更新
kasmweb/ubuntu-noble-nvidia-pytorch
kasmweb
暂无描述
1 次收藏1万+ 次下载
2 小时前更新
镜像拉取常见问题
功能
排错
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"