rocm/jax Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务
rocm/jaxrocm
JAX针对ROCm后端的Docker镜像，包含JAX在ROCm框架中的最新支持
5 次收藏下载次数: 0状态：社区镜像维护者：rocm仓库类型：镜像最近更新：1 个月前
JAX-ROCm Docker镜像
本仓库包含JAX在ROCm框架中的最新支持。
镜像说明
JAX-ROCm Docker镜像提供了JAX在ROCm平台上的运行环境，整合了JAX针对ROCm后端的最新功能与优化，支持基于AMD ROCm加速的高性能计算任务。
使用场景
适用于需要在ROCm架构下运行JAX相关应用的场景，如深度学习模型训练、科学计算等依赖GPU加速的任务。
部署方案示例
推荐通过以下命令启动镜像：
-
创建便捷命令别名：
alias drun='sudo docker run -it --network=host --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -w /root' -
启动镜像：
drun rocm/jax:latest
镜像拉取常见问题
使用与功能问题
错误码与失败问题
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访问体验非常流畅，大镜像也能快速完成下载。"