JAX/Flaxの実行速度 [M2 Mac, CPU, GPU]
YK
Yuki Kashiwada / December 13, 2022
4 min read
結果
- M2 Mac: 224.94 秒
- CPU: 258.22 秒
- GPU: 44.70 秒
ということで、GPU は M2 Mac, CPU の 5 倍も速かったです。また、M2 Mac が第 10 世代の Core i5(以下参照)よりも 10%ほど速い結果となりました。
スペックの詳細
- CPU: Intel(R) Core(TM) i5-10400 CPU @ 2.90GHz
- GPU: NVIDIA GeForce RTX 3090
M2 MacBook Air 13 inch
None
CPU on Ubuntu
None
GPU
None