概要
- フル名
docker.io/modelbox/modelbox-develop-tensorflow_2.6.0-cuda_11.2-ubuntu-x86_64- Registry
- docker.io
- 名前空間
- modelbox
- リポジトリ
- modelbox-develop-tensorflow_2.6.0-cuda_11.2-ubuntu-x86_64
- キャッシュ Tag 数
- 12
- 最終同期
- 07/06/2026, 06:45 PM
概要
ModelBox TensorFlow 2.6 + CUDA 11.2 开发镜像,预装推理框架与 GPU 工具链,用于视频 AI 流水线本地调试与自定义 flowunit 编译。
詳細
本镜像面向华为 ModelBox 生态,集成 TensorFlow 2.6、CUDA 11.2 与 ModelBox SDK,供开发者编写 .toml 图与 C++/Python flowunit。必须 NVIDIA 驱动与 --gpus all,镜像体积大且偏交互开发,不适合无 GPU 的 Kubernetes Deployment 长期副本。正确用法是在 GPU 工作站或带 nvidia.com/gpu 的 Job 中挂载工程目录编译调试,生产推理请用 ModelBox 运行时专用镜像。
快速启动
docker run --rm -it --gpus all \
-v $(pwd):/workspace \
docker.io/modelbox/modelbox-develop-tensorflow_2.6.0-cuda_11.2-ubuntu-x86_64:latest \
bash
推荐实践
docker run --rm --gpus all \
-v $(pwd):/workspace -w /workspace \
-e CUDA_VISIBLE_DEVICES=0 \
docker.io/modelbox/modelbox-develop-tensorflow_2.6.0-cuda_11.2-ubuntu-x86_64:1.0.0 \
modelbox-tool build -p .
核心参数说明
--gpus all— 启用 CUDA 开发环境-v $(pwd):/workspace— 挂载 flow 工程-e CUDA_VISIBLE_DEVICES— 指定 GPU 编号-w /workspace— 构建工作目录--rm— 开发会话结束释放资源- 用途 — 开发编译,非生产 Serving 镜像