rocm/pytorch
rocm
基于ROCm的PyTorch Docker镜像为开发者提供了预配置的深度学习环境，集成了PyTorch框架与AMD ROCm开源计算平台，支持AMD GPU硬件加速，无需手动配置驱动及库依赖，可直接用于深度学习模型的开发、训练与部署，有效简化环境搭建流程，确保跨平台一致性，适用于科研机构、工业界等多种场景，助力高效利用AMD硬件资源开展AI相关任务。
121 次收藏下载次数: 0状态：社区镜像维护者：rocm仓库类型：镜像最近更新：12 天前
ROCm后端PyTorch Docker镜像使用说明
本仓库提供基于ROCm后端的PyTorch版本Docker镜像。
1. 支持的GPU型号
仓库中的Docker镜像可在以下GPU型号上运行：
- gfx908（对应MI100）
- gfx90a（对应MI210/MI250/MI250x）
- gfx942（对应MI300A/MI300X/MI325）
- gfx950（对应MI350/MI355，自ROCm 7.0起支持）
- gfx1030（基于Navi21的SKU型号）
- gfx1100/gfx1101（基于Navi31的SKU型号）
- gfx1200/gfx1201（基于Navi44/Navi48的SKU型号，自ROCm 6.4.2起支持）
2. 配置Docker环境
如需配置支持ROCm的Docker环境，操作指引请参考：
ROCm Docker环境快速配置指南
3. 运行rocm/pytorch Docker镜像
3.1 添加快捷命令别名
为简化拉取和运行PyTorch容器的命令，建议将以下别名添加到.profile或.bashrc文件中（添加后需通过source ~/.bashrc或重启终端生效）：
bashalias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --shm-size 8G -v $HOME/dockerx:/dockerx -w /dockerx'
3.2 启动容器示例
运行最新版本的ROCm PyTorch容器，执行以下命令即可：
bashdrun rocm/pytorch
或显式指定latest标签：
bashdrun rocm/pytorch:latest
更多相关 Docker 镜像与资源
以下是 rocm/pytorch 相关的常用 Docker 镜像，适用于 不同场景 等不同场景：
- pytorch/pytorch Docker 镜像说明
- bitnami/pytorch Docker 镜像说明（PyTorch 深度学习框架，Bitnami 企业级配置）
- nvidia/cuda Docker 镜像说明
- dustynv/vllm Docker 镜像说明（高性能 LLM 推理服务）
- rocm/vllm Docker 镜像说明（vLLM 推理服务，ROCm GPU 优化版本）
镜像拉取方式
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问 版本下载页面。
更多 pytorch 镜像推荐
pytorch/pytorch
pytorch
PyTorch是一款以Python为首要设计理念的深度学习框架，凭借简洁易用的Python接口、动态计算图机制及强大的灵活性，广泛应用于学术研究与工业开发，支持从快速原型设计到大规模部署的全流程，深度融合Python数据科学生态，为开发者提供高效且直观的深度学习解决方案。
1.6千 次收藏1000万+ 次下载
19 天前更新
bitnami/pytorch
Bitnami Secure Images（VMware Tanzu）
比特纳米PyTorch安全镜像是一款为深度学习框架PyTorch量身打造的预配置、安全加固型容器镜像，集成经过严格测试的依赖组件，具备漏洞扫描、合规性检查及持续更新机制，可有效保障开发环境安全，简化从模型训练到部署的全流程，适用于科研机构、企业开发者在AI项目中快速构建稳定、安全的PyTorch运行环境。
77 次收藏100万+ 次下载
19 天前更新
bitnamicharts/pytorch
bitnamicharts
Bitnami提供的PyTorch Helm chart，用于在Kubernetes环境中简化PyTorch的部署与管理。
10万+ 次下载
19 天前更新
chainguard/pytorch
chainguard
使用Chainguard的低至零CVE容器镜像构建、交付和运行安全软件。
1万+ 次下载
9 天前更新
openeuler/pytorch
openeuler
基于openEuler构建的PyTorch官方Docker镜像，提供张量计算（支持GPU加速）和基于自动微分系统的深度神经网络功能，适用于深度学习开发与研究。
1 次收藏309 次下载
2 年前更新
pytorch/almalinux-builder
pytorch
已安装conda的通用目的镜像，用于PyTorch的CI/CD流程。
100万+ 次下载
9 天前更新
轩辕镜像配置手册
探索更多轩辕镜像的使用方法，找到最适合您系统的配置方式
Docker 配置
登录仓库拉取
通过 Docker 登录认证访问私有仓库
专属域名拉取
无需登录使用专属域名
K8s Containerd
Kubernetes 集群配置 Containerd
K3s
K3s 轻量级 Kubernetes 镜像加速
Dev Containers
VS Code Dev Containers 配置
Podman
Podman 容器引擎配置
Singularity/Apptainer
HPC 科学计算容器配置
其他仓库配置
ghcr、Quay、nvcr 等镜像仓库
Harbor 镜像源配置
Harbor Proxy Repository 对接专属域名
Portainer 镜像源配置
Portainer Registries 加速拉取
Nexus 镜像源配置
Nexus3 Docker Proxy 内网缓存
系统配置
需要其他帮助？请查看我们的 常见问题Docker 镜像访问常见问题解答 或 提交工单
镜像拉取常见问题
使用与功能问题
错误码与失败问题
docker pull 提示 manifest unknown 怎么办？
manifest unknown
docker pull 提示 no matching manifest 怎么办？
no matching manifest（架构）
镜像已拉取完成，却提示 invalid tar header 或 failed to register layer 怎么办？
invalid tar header（解压）
Docker pull 时 HTTPS / TLS 证书验证失败怎么办？
TLS 证书失败
Docker pull 时 DNS 解析超时或连不上仓库怎么办？
DNS 超时
Docker 拉取出现 410 Gone 怎么办？
410 Gone 排查
出现 402 或「流量用尽」提示怎么办？
402 与流量用尽
Docker 拉取提示 UNAUTHORIZED（401）怎么办？
401 认证失败
遇到 429 Too Many Requests（请求太频繁）怎么办？
429 限流
docker login 提示 Cannot autolaunch D-Bus，还算登录成功吗？
D-Bus 凭证提示
为什么会出现「单层超过 20GB」或 413，无法加速拉取？
413 与超大单层
账号 / 计费 / 权限
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"