
如果你使用 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/pytorch:1.2-cuda10.0-cudnn7-runtime构建,集成了UNET和R2UNET两种经典深度学习图像分割算法,提供开箱即用的图像分割任务运行环境。镜像支持GPU加速计算,适用于需要高效执行图像分割任务的场景。
bashdocker pull docker.xuanyuan.run/[镜像名称]:[标签] # 请替换为实际镜像名称和标签
使用CPU运行
bashdocker run -it --rm \ -v /本地数据目录:/app/data \ [镜像名称]:[标签] \ python /app/segmentation.py --model unet --input /app/data/input.jpg --output /app/data/output.jpg
使用GPU运行
bashdocker run -it --rm --gpus all \ -v /本地数据目录:/app/data \ [镜像名称]:[标签] \ python /app/segmentation.py --model r2unet --input /app/data/input.jpg --output /app/data/output.jpg
| 参数 | 说明 | 可选值 | 默认值 |
|---|---|---|---|
--model | 选择分割模型 | unet, r2unet | unet |
--input | 输入图像路径 | 本地或容器内路径 | 无 |
--output | 输出分割结果路径 | 本地或容器内路径 | 无 |
--device | 计算设备 | cpu, cuda | 自动检测 |
yamlversion: '3' services: segmentation: image: [镜像名称]:[标签] runtime: nvidia # 若使用GPU需添加 volumes: - ./data:/app/data command: python /app/segmentation.py --model r2unet --input /app/data/input.jpg --output /app/data/output.jpg
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本,请访问 标签列表页面。
来自真实用户的反馈,见证轩辕镜像的优质服务
以下是 eperezp1990/pytorch 相关的常用 Docker 镜像,适用于 不同场景 等不同场景: