pytorch-TPU_test
一个mnist实验在CPU、GPU、TPU上的速度测试
- https://colab.research.google.com/drive/10-Wx9tXDEQzaIz2AFlG1_h_81emIEbDG 感觉TPU的性能还是不如GPU,网上很多文章总说TPU速度可以碾压GPU......
效果 | |
---|---|
CPU | 35s/it |
GPU | 9s/it |
TPU | 24s/it |
感激
该测试参考了
一个mnist实验在CPU、GPU、TPU上的速度测试
效果 | |
---|---|
CPU | 35s/it |
GPU | 9s/it |
TPU | 24s/it |
该测试参考了