ubuntu 中tensorflow-gpu性能测试

前言

折腾了小半个星期,终于把ubuntu环境下的gpu版的tensorflow给弄好了,在一段卷积神经网络的程序中进行了测试,速度提升了四倍左右,可能因程序而异,也可能就是因为我这显卡太次,提升的不多,但总算是有了质的改变,下面测试以下。

查看cpu以及gpu实时使用情况

在这里插入图片描述

输入以后,回车,得到下图,watch -n 1 代表每一秒刷新一次使用情况。
在这里插入图片描述
新打开一个终端,输入htop,如果没有安装的话,先安装htop,这是一个查看cpu使用情况的软件,如下图:
在这里插入图片描述

测试

测试代码可以私信博主,下面是具体效果:
这是gpu版的:
在这里插入图片描述

在运行过程中cpu和gpu的使用大致情况如下:
在这里插入图片描述可以看到,gpu的使用率以下飙到了百分之70多,cpu的使用率只有小幅的增加。

这是cpu版的:
在这里插入图片描述

使用过程中cpu和gpu的使用情况如下:

在这里插入图片描述可以看到,cpu使用率大幅增加,GPU小幅增加。一个耗时20s,一个是81s,速度提升了四倍。

结语

事实证明,在大规模的并行运算中,特别是在CV领域,gpu体现出了巨大的优势。

猜你喜欢

转载自blog.csdn.net/qq_29566629/article/details/89313198
今日推荐