pytorch cpu gpu性能对比

经过同样的模型预测对比

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