从零点一开始机器学习之GPU运算性能和CPU性能对比(GPU算力表)

741 阅读1分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第1天,点击查看活动详情

很多新入门的小伙伴和我一样,都知道机器学习需要GPU去跑,CPU性能不好,但是到底有多大的差距呢?刚开始的同学没有一个直观的认识(主要是我自己也没有直观的认识),在此记录一下吧。 我们采用同样的一个简单的基于VGG16的图像分类代码来对比一下,如果有误导或者理解不到位的地方,恳请指正,谢谢!服务器虽然有点旧,但是还是可以看出问题的! 图像分三类,每一类图像3800张左右,batch_size均为25(最好为2的次方数)


CPU服务器一


  • 服务器配置 image.png
  • 选取epoch5-14查看计算用时

image.png

CPU服务器二


  • 服务器配置 image.png
  • 选取epoch5-14查看计算用时

image.png

GPU计算机一


  • 服务器配置

image.png

  • 显卡规格 在这里插入图片描述

  • 选取epoch5-14查看计算用时 在这里插入图片描述 在用gpu进行计算的时候,观察cpu也有40%左右的占用率,不是说好用gpu计算的吗?应该是cpu也进行了一些TensorFlow其他的计算吧。 可以看出一张算力6.1的1050卡,基本上是10+倍8核E5cpu的性能。

  • 后记 还有一个GT730的卡,辛辛苦苦花了很多时间装好了环境,一运行居然告诉我算力只有3.5,cuda最低要求为3.7,新手同学可以查询一下显卡算力免得浪费功夫装很久环境!Cuda compute capability 3.5. The minimum required Cuda capability is 3.7。

  • GPU算力表 https://developer.nvidia.com/cuda-gpus 在这里插入图片描述 在这里插入图片描述