cpu消耗 pytorch_Pytorch用GPU到底能比CPU快多少？

作者：AllinToyou | 2024-05-30 00:20:17

踩

cupy和pytorch哪个快

给踌躇于要不要买GPU的朋友们做一点微小的贡献：

同一段脚本，同样的数据量，同样的神经网络配置，用cpu和gpu分别计算，看看分别用了多长时间。

嫌麻烦的同学可以提前看结论：

同样的钱，买GPU得到的计算能力，是CPU的15倍。

15倍。

神经网络配置：

5层hidden layer，每层500个nodes，一共500个epochs，做一次实验;

5层hidden layer，每层1000个nodes，一共1000个epochs，再做一次实验;

CPU信息：

4 Intel(R) Core(TM) i5-6600K CPU @ 3.50GHz

4核的inteli5-6600K处理器，主频3.50GHz。目前市场价大约250刀。

如图所示，确实4个cpu核心都用上了，都在干活儿。

GPU信息：

NVIDIA GeForce GTX 1070 8GB

一块1070的GPU。我用的是的1070，550刀; 假如是业界挖矿明星1080Ti，应该会更快，1080Ti目前大约95

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】