前言
折腾了小半个星期,终于把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体现出了巨大的优势。