
Ivy 是一款专注于跨机器学习框架模型转换的工具,旨在解决不同深度学习框架(如PyTorch、Tensorflow、JAX等)之间的模型格式兼容性问题。该Docker镜像封装了Ivy工具的完整运行环境,提供开箱即用的模型转换能力,帮助开发者和研究人员高效实现跨框架模型迁移。
bash# 拉取镜像(假设镜像名为 ivytool/ivy) docker pull ivytool/ivy # 转换PyTorch模型到Tensorflow格式 docker run --rm -v $(pwd):/workspace ivytool/ivy \ ivy convert \ --source pytorch \ --target tensorflow \ /workspace/source_model.pt \ /workspace/target_model.h5
| 参数 | 描述 | 可选值 |
|---|---|---|
--source | 源框架类型 | pytorch, tensorflow, jax |
--target | 目标框架类型 | pytorch, tensorflow, jax |
--device | 转换使用的设备 | cpu, cuda(需宿主机支持) |
--precision | 转换精度 | float32, float64 |
bashdocker run --rm -v $(pwd):/workspace ivytool/ivy \ ivy convert \ --source jax \ --target pytorch \ --device cpu \ /workspace/jax_model.pkl \ /workspace/pytorch_model.pt
-v 参数正确挂载--device cuda 加速转换过程(需Docker支持GPU)您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本,请访问 标签列表页面。



探索更多轩辕镜像的使用方法,找到最适合您系统的配置方式
通过 Docker 登录认证访问私有仓库
无需登录使用专属域名
Kubernetes 集群配置 Containerd
K3s 轻量级 Kubernetes 镜像加速
VS Code Dev Containers 配置
Podman 容器引擎配置
HPC 科学计算容器配置
ghcr、Quay、nvcr 等镜像仓库
Harbor Proxy Repository 对接专属域名
Portainer Registries 加速拉取
Nexus3 Docker Proxy 内网缓存
需要其他帮助?请查看我们的 常见问题Docker 镜像访问常见问题解答 或 提交工单
docker search 限制
站内搜不到镜像
离线 save/load
插件要用 plugin install
WSL 拉取慢
安全与 digest
新手拉取配置
镜像合规机制
manifest unknown
no matching manifest(架构)
invalid tar header(解压)
TLS 证书失败
DNS 超时
域名连通性排查
410 Gone 排查
402 与流量用尽
401 认证失败
429 限流
D-Bus 凭证提示
413 与超大单层
来自真实用户的反馈,见证轩辕镜像的优质服务