dustynv/tensorflow2

tensorflow Docker镜像文档

镜像概述和主要用途

tensorflow镜像是支持TensorFlow 1.x（TF1）和TensorFlow 2.x（TF2）的Docker容器，具备CUDA加速能力，专为Jetson平台设计。该镜像基于NVIDIA官方Jetson平台TensorFlow轮子构建，适用于深度学习模型的部署与运行。与https://github.com/dusty-nv/jetson-containers/tree/master/packages/l4t/l4t-tensorflow%E5%AE%B9%E5%99%A8%E7%9B%B8%E6%AF%94%EF%BC%8C%E6%9C%AC%E9%95%9C%E5%83%8F%E4%B8%8D%E5%90%ABOpenCV%E5%92%8CPyCUDA%EF%BC%8C%E4%B8%93%E6%B3%A8%E4%BA%8E%E6%A0%B8%E5%BF%83TensorFlow%E5%8A%9F%E8%83%BD%E3%80%82

核心功能和特性

容器信息

镜像信息

使用场景和适用范围

• 适用平台：Jetson系列设备（如Jetson Nano、Jetson Xavier NX、Jetson AGX Xavier等）
• 系统要求：JetPack 4.6+（L4T R32.6+）或JetPack 5.1+（L4T R35.x+）
• 典型用途：深度学习模型推理部署、TensorFlow应用开发与测试、依赖CUDA加速的AI任务执行

使用方法和配置说明

运行容器

可通过https://github.com/dusty-nv/jetson-containers/tree/master/docs/run.md%E5%B7%A5%E5%85%B7%E7%BB%93%E5%90%88https://github.com/dusty-nv/jetson-containers/tree/master/docs/run.md#autotag%E8%87%AA%E5%8A%A8%E9%80%89%E6%8B%A9%E5%85%BC%E5%AE%B9%E9%95%9C%E5%83%8F%EF%BC%8C%E6%88%96%E6%89%8B%E5%8A%A8%E4%BD%BF%E7%94%A8%60docker run`命令启动容器：

bash
# 自动拉取或构建兼容的容器镜像
jetson-containers run $(autotag tensorflow)

# 或显式指定上述镜像列表中的某个镜像
jetson-containers run dustynv/tensorflow:r32.7.1

# 若使用`docker run`（需指定镜像及挂载等参数）
sudo docker run --runtime nvidia -it --rm --network=host dustynv/tensorflow:r32.7.1

^{https://github.com/dusty-nv/jetson-containers/tree/master/docs/run.md%E4%BC%9A%E5%B0%86%E5%8F%82%E6%95%B0%E8%BD%AC%E5%8F%91%E7%BB%99%60docker run，并添加默认配置（如--runtime nvidia、挂载/data`缓存、设备检测等）
https://github.com/dusty-nv/jetson-containers/tree/master/docs/run.md#autotag%E4%BC%9A%E6%9F%A5%E6%89%BE%E4%B8%8E%E5%BD%93%E5%89%8DJetPack/L4T%E7%89%88%E6%9C%AC%E5%85%BC%E5%AE%B9%E7%9A%84%E5%AE%B9%E5%99%A8%E9%95%9C%E5%83%8F%EF%BC%88%E6%9C%AC%E5%9C%B0%E9%95%9C%E5%83%8F%E3%80%81%E8%BF%9C%E7%A8%8B%E4%BB%93%E5%BA%93%E6%8B%89%E5%8F%96%E6%88%96%E8%87%AA%E5%8A%A8%E6%9E%84%E5%BB%BA%EF%BC%89}

挂载本地目录

使用-v或--volume参数将主机目录挂载到容器中：

bash
jetson-containers run -v /主机路径:/容器路径 $(autotag tensorflow)

运行命令而非交互式shell

启动容器时直接执行指定命令：

bash
jetson-containers run $(autotag tensorflow) 应用程序 --参数 数值

构建容器

若使用上述autotag工具，当本地无兼容镜像时会自动提示构建。也可手动构建容器，步骤如下：

1. 先完成https://github.com/dusty-nv/jetson-containers/tree/master/docs/setup.md
2. 执行构建命令：

bash
jetson-containers build tensorflow

构建过程中会自动包含所需依赖项并进行测试。可通过--help查看更多构建选项：

bash
jetson-containers build tensorflow --help