tensorflow-gpu版本安装及深度神经网络训练与cpu版本对比( 二 )


文章插图
GPU耗时:

tensorflow-gpu版本安装及深度神经网络训练与cpu版本对比

文章插图
切换CPU GPU 只要切换设备就行了,我只进行了2epoch的卷积训练,可以看到GPU速度要比CPU快个4 5 倍左右,如果是前馈神经网络或者简单的神经网络,测试验证使用CPU是被GPU要快的,所以自己需要根据实际情况切换设备 。
需要zlib文件的可以给我留言 。

经验总结扩展阅读