pytorch cpu gpu性能对比

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jacke121/article/details/82049035

经过同样的模型预测对比

gpu比cpu快38倍左右。

gpu需要16ms

0 863001
1 22000
2 15001
3 16999
4 18000
5 15000
6 17000
7 16999
8 17001
9 16000

cpu需要610ms

0 601998
1 610000
2 611999
3 612000
4 633033
5 623966

猜你喜欢

转载自blog.csdn.net/jacke121/article/details/82049035