tencentos/tencentos4-tensorflow repository overview

⁠镜像信息

• 基础镜像: TencentOS4 + CUDA 12.9
• 深度学习框架: TensorFlow 2.20.0 GPU 版
• 默认工作目录: /workspace
• 内置示例: tensorflow_demo.py (MNIST 训练脚本)

⁠快速开始

⁠1. 拉取并启动容器

执行以下命令从 Docker Hub 拉取镜像并启动容器：

sudo docker run -itd --privileged --gpus all --name=tencentos4-tensorflow tencentos/tencentos4-tensorflow

参数说明:

• --gpus all: 使容器可以访问所有 GPU 设备
• --privileged: 赋予容器扩展权限
• --name: 指定容器名称为 tencentos4-tensorflow

⁠2. 验证容器状态

启动后，使用以下命令查看容器运行状态：

sudo docker ps

⁠3. 进入容器

执行以下命令进入运行中的容器：

sudo docker exec -it tencentos4-tensorflow bash

⁠4. 运行示例程序

进入容器后，默认位于 /workspace 目录。执行以下命令运行 MNIST 训练示例：

python tensorflow_demo.py

该脚本将训练一个简单的全连接神经网络模型，用于识别 MNIST 手写数字数据集。

⁠注意事项

• 确保宿主机已安装 NVIDIA 驱动和 Docker GPU 支持
• 运行示例程序首次运行时会自动下载 MNIST 数据集
• 运行示例程序默认训练 5 个 epoch，可根据需要修改脚本参数