热门搜索:
cemizm/tf-benchmark-gpu
cemizm
这是一个支持GPU加速的Docker化TensorFlow基准测试工具镜像，可运行ResNet50、AlexNet等图像分类模型的性能测试，通过命令行参数配置GPU数量、模型类型等参数。
3 次收藏下载次数: 0状态：社区镜像维护者：cemizm仓库类型：镜像最近更新：5 年前
让 AI 帮你使用轩辕镜像？ · 展开查看说明
如果你使用 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 无法访问外链，可 打开说明文档 复制全文粘贴。文档会随站点更新，复制内容可能过期，建议定期检查。
支持GPU的Docker化TensorFlow基准测试工具
本Docker镜像基于最新的https://hub.docker.com/r/tensorflow/tensorflow GPU版本镜像构建，集成了https://github.com/tensorflow/benchmarks%E4%BB%93%E5%BA%93%E4%BD%9C%E4%B8%BA%E5%AE%B9%E5%99%A8%E5%85%A5%E5%8F%A3%E7%82%B9%EF%BC%8C%E5%8F%AF%E9%80%9A%E8%BF%87%E5%91%BD%E4%BB%A4%E8%A1%8C%E5%8F%82%E6%95%B0%E5%9C%A8GPU%E7%8E%AF%E5%A2%83%E4%B8%AD%E8%BF%90%E8%A1%8C%E5%A4%9A%E7%A7%8D%E5%9B%BE%E5%83%8F%E5%88%86%E7%B1%BB%E6%A8%A1%E5%9E%8B%E7%9A%84%E6%80%A7%E8%83%BD%E5%9F%BA%E5%87%86%E6%B5%8B%E8%AF%95%E3%80%82
更多信息请参考官方TensorFlow指南。
运行基准测试
若要使用单GPU运行ResNet50模型（基于合成数据），执行以下命令：
bashdocker run --runtime=nvidia --rm cemizm/tf-benchmark-gpu --model resnet50 --num_gpus=1
常用参数
- model：需测试的模型，例如alexnet、resnet50、resnet152、inception3、vgg16，默认值为trivial。
- num_gpus：使用的GPU数量，默认使用所有可用GPU。
- variable_update：变量管理方式，可选parameter_server、replicated、distributed_replicated、independent，默认值为parameter_server。
- batch_size：每个GPU的批次大小，默认值为32。
查看所有可用参数：
bashdocker run --runtime=nvidia --rm cemizm/tf-benchmark-gpu --help
测试结果
官方TensorFlow在P100和K80上的测试结果可参考上述链接。以下是工作站环境中消费级硬件的测试结果：
| 设置项 | 值 |
|---|---|
| TensorFlow | 1.14 |
| 数据集 | imagenet（合成） |
| 模式 | 训练 |
| SingleSess | False |
| 批次数量 | 100 |
| epoch数量 | 0.00 |
| NUMA绑定 | False |
| 数据格式 | NCHW |
| 优化器 | sgd |
| 变量管理方式 | parameter_server |
RTX 2070
| GPU数量 | InceptionV3 | ResNet-50 | ResNet-152 | AlexNet | VGG16 |
|---|---|---|---|---|---|
| 1 | 122 | 196 | 76 | 2371 | 112 |
| 2 | 209 | 364 | 133 | 4408 | 173 |
| 4 | |||||
| 8 |
RTX 2070 Super
| GPU数量 | InceptionV3 | ResNet-50 | ResNet-152 | AlexNet | VGG16 |
|---|---|---|---|---|---|
| 1 | 134 | 214 | 84 | 2610 | 124 |
| 2 | |||||
| 4 | |||||
| 8 |
RTX 2080 TI
| GPU数量 | InceptionV3 | ResNet-50 | ResNet-152 | AlexNet | VGG16 |
|---|---|---|---|---|---|
| 1 | 194 | 308 | 122 | 3982 | 186 |
| 2 | |||||
| 4 | |||||
| 8 |
镜像拉取方式
您可以使用以下命令拉取该镜像。请将 <标签> 替换为具体的标签版本。如需查看所有可用标签版本，请访问 标签列表页面。
轩辕镜像加速拉取命令点我查看更多 tf-benchmark-gpu 镜像标签
docker pull docker.xuanyuan.run/cemizm/tf-benchmark-gpu:<标签>
DockerHub 原生拉取命令
docker pull cemizm/tf-benchmark-gpu:<标签>
更多 tf-benchmark-gpu 镜像推荐
redislabs/memtier_benchmark
redislabs
用于运行memtier_benchmark的Docker镜像，这是一款由Redis开发的NoSQL键值数据库负载生成和基准测试工具，支持Redis/Memcache协议、多线程执行及多种配置选项。
1 次收藏100万+ 次下载
9 天前更新
hashicorp/vault-benchmark
hashicorp
暂无描述
5万+ 次下载
2 年前更新
owasp/benchmark
owasp
运行OWASP Benchmark的最新版本，自动拉取、构建并运行OWASP BenchmarkJava项目，遵循官方快速启动指南，监听8443端口供外部访问。
14 次收藏10万+ 次下载
5 个月前更新
vdaas/vald-benchmark-job
vdaas
vald-benchmark-job镜像是用于vald-benchmark-job组件的Docker镜像。
5万+ 次下载
7 天前更新
opensearchproject/opensearch-benchmark
opensearchproject
OpenSearch-Benchmark是OpenSearch的宏基准测试框架，主要用于运行基准测试。
3 次收藏5万+ 次下载
25 天前更新
redis/vector-db-benchmark
Redis 官方镜像
用于向量搜索引擎基准测试的框架，支持Redis（含RediSearch和Vector Sets）、Weaviate、Milvus等多种向量数据库，可分析精度与性能权衡，提供实时监控和详细结果报告。
1 次收藏2.3千+ 次下载
16 天前更新
镜像拉取常见问题
功能
错误码
用户好评
来自真实用户的反馈，见证轩辕镜像的优质服务
oldzhang
运维工程师
Linux服务器
5
"Docker访问体验非常流畅，大镜像也能快速完成下载。"