rocm/pytorch Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务
rocm/pytorchrocm
基于ROCm的PyTorch Docker镜像为开发者提供了预配置的深度学习环境，集成了PyTorch框架与AMD ROCm开源计算平台，支持AMD GPU硬件加速，无需手动配置驱动及库依赖，可直接用于深度学习模型的开发、训练与部署，有效简化环境搭建流程，确保跨平台一致性，适用于科研机构、工业界等多种场景，助力高效利用AMD硬件资源开展AI相关任务。
120 次收藏下载次数: 0状态：社区镜像维护者：rocm仓库类型：镜像最近更新：1 个月前
This repo hosts the docker images for PyTorch releases with ROCm backend support.
(1) Supported GPUs:
The docker images hosted in this registry will run on:
- gfx908 (MI100)
- gfx90a (MI210/MI250/MI250x)
- gfx942 (MI300A/MI300X/MI325)
- (as of ROCm 7.0) gfx950 (MI350/MI355)
- gfx1030 (Navi21-based SKUs)
- gfx1100/gfx1101 (Navi31-based SKUs)
- (as of ROCm 6.4.2) gfx1200/gfx1201 (Navi44/Navi48-based SKUs)
- (as of ROCm 7.1) gfx1151/gfx1150 (Strix Halo/Strix Point-based SKUs)
(2) Install Docker Environment
To configure docker environment for ROCm, please refer to the following installation guide: [***]
(3) Run the rocm/pytorch Docker Image:
To facilitate commands to pull and run the latest PyTorch docker container, add the following alias to your .profile or .bashrc:
alias 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'
For example, to get the latest PyTorch on ROCm, run the following command:
drun rocm/pytorch
or
drun rocm/pytorch:latest
pytorch/pytorch
pytorch
PyTorch是一款以Python为首要设计理念的深度学习框架，凭借简洁易用的Python接口、动态计算图机制及强大的灵活性，广泛应用于学术研究与工业开发，支持从快速原型设计到大规模部署的全流程，深度融合Python数据科学生态，为开发者提供高效且直观的深度学习解决方案。
1.6千 次收藏1000万+ 次下载
1 个月前更新
bitnami/pytorch
bitnami
比特纳米PyTorch安全镜像是一款为深度学习框架PyTorch量身打造的预配置、安全加固型容器镜像，集成经过严格测试的依赖组件，具备漏洞扫描、合规性检查及持续更新机制，可有效保障开发环境安全，简化从模型训练到部署的全流程，适用于科研机构、企业开发者在AI项目中快速构建稳定、安全的PyTorch运行环境。
77 次收藏100万+ 次下载
22 天前更新
bitnamicharts/pytorch
bitnamicharts
Bitnami提供的PyTorch Helm chart，用于在Kubernetes环境中简化PyTorch的部署与管理。
10万+ 次下载
15 天前更新
chainguard/pytorch
chainguard
使用Chainguard的低至零CVE容器镜像构建、交付和运行安全软件。
1万+ 次下载
13 天前更新
pytorch/almalinux-builder
pytorch
已安装conda的通用目的镜像，用于PyTorch的CI/CD流程。
100万+ 次下载
13 天前更新
pytorch/torchserve
pytorch
PyTorch Serve Docker镜像是用于部署PyTorch模型的服务框架，提供REST API接口，支持模型管理、推理及扩展，简化生产环境中PyTorch模型的部署流程。
32 次收藏100万+ 次下载
1 年前更新
镜像拉取常见问题
使用与功能问题
错误码与失败问题
manifest unknown 错误：镜像不存在或标签错误
manifest unknown 错误
TLS/SSL 证书验证失败：Docker pull 时 HTTPS 证书错误
TLS 证书验证失败
DNS 解析超时：无法解析镜像仓库地址或连接超时
DNS 解析超时
410 Gone 错误：Docker 版本过低导致协议不兼容
410 错误：版本过低
402 Payment Required 错误：流量耗尽错误提示
402 错误：流量耗尽
401 UNAUTHORIZED 错误：身份认证失败或登录信息错误
身份认证失败错误
429 Too Many Requests 错误：请求频率超出专业版限制
429 限流错误
Docker login 凭证保存错误：Cannot autolaunch D-Bus（不影响登录）
凭证保存错误
账号 / 计费 / 权限
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"