
# 結果

- 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

```
train epoch: 1, loss: 0.1405, accuracy: 95.81
test epoch: 1, loss: 0.05, accuracy: 98.22
Epoch 1 in 22.32 sec
train epoch: 2, loss: 0.0502, accuracy: 98.44
test epoch: 2, loss: 0.04, accuracy: 98.66
Epoch 2 in 21.22 sec
train epoch: 3, loss: 0.0343, accuracy: 98.94
test epoch: 3, loss: 0.03, accuracy: 98.97
Epoch 3 in 21.59 sec
train epoch: 4, loss: 0.0259, accuracy: 99.16
test epoch: 4, loss: 0.04, accuracy: 99.00
Epoch 4 in 21.94 sec
train epoch: 5, loss: 0.0198, accuracy: 99.42
test epoch: 5, loss: 0.04, accuracy: 98.93
Epoch 5 in 22.07 sec
train epoch: 6, loss: 0.0180, accuracy: 99.42
test epoch: 6, loss: 0.05, accuracy: 98.48
Epoch 6 in 21.78 sec
train epoch: 7, loss: 0.0153, accuracy: 99.53
test epoch: 7, loss: 0.05, accuracy: 98.60
Epoch 7 in 21.23 sec
train epoch: 8, loss: 0.0133, accuracy: 99.59
test epoch: 8, loss: 0.05, accuracy: 98.93
Epoch 8 in 21.77 sec
train epoch: 9, loss: 0.0108, accuracy: 99.65
test epoch: 9, loss: 0.05, accuracy: 99.00
Epoch 9 in 21.39 sec
train epoch: 10, loss: 0.0112, accuracy: 99.66
test epoch: 10, loss: 0.04, accuracy: 99.16
Epoch 10 in 21.76 sec
Total in 224.94 sec
```

## CPU on Ubuntu

```
train epoch: 1, loss: 0.1424, accuracy: 95.62
test epoch: 1, loss: 0.06, accuracy: 98.24
Epoch 1 in 26.10 sec
train epoch: 2, loss: 0.0485, accuracy: 98.51
test epoch: 2, loss: 0.05, accuracy: 98.53
Epoch 2 in 24.67 sec
train epoch: 3, loss: 0.0362, accuracy: 98.88
test epoch: 3, loss: 0.03, accuracy: 99.10
Epoch 3 in 24.72 sec
train epoch: 4, loss: 0.0253, accuracy: 99.21
test epoch: 4, loss: 0.04, accuracy: 98.99
Epoch 4 in 24.70 sec
train epoch: 5, loss: 0.0230, accuracy: 99.31
test epoch: 5, loss: 0.03, accuracy: 99.15
Epoch 5 in 24.69 sec
train epoch: 6, loss: 0.0169, accuracy: 99.50
test epoch: 6, loss: 0.04, accuracy: 98.81
Epoch 6 in 24.65 sec
train epoch: 7, loss: 0.0149, accuracy: 99.53
test epoch: 7, loss: 0.03, accuracy: 99.09
Epoch 7 in 24.68 sec
train epoch: 8, loss: 0.0130, accuracy: 99.59
test epoch: 8, loss: 0.05, accuracy: 98.63
Epoch 8 in 24.73 sec
train epoch: 9, loss: 0.0118, accuracy: 99.63
test epoch: 9, loss: 0.05, accuracy: 98.93
Epoch 9 in 24.71 sec
train epoch: 10, loss: 0.0101, accuracy: 99.69
test epoch: 10, loss: 0.04, accuracy: 99.14
Epoch 10 in 24.71 sec
Total in 258.22 sec
```

## GPU

```
train epoch: 1, loss: 0.0107, accuracy: 99.67
test epoch: 1, loss: 0.03, accuracy: 98.99
Epoch 1 in 4.53 sec
train epoch: 2, loss: 0.0075, accuracy: 99.79
test epoch: 2, loss: 0.04, accuracy: 98.98
Epoch 2 in 4.49 sec
train epoch: 3, loss: 0.0108, accuracy: 99.67
test epoch: 3, loss: 0.04, accuracy: 98.99
Epoch 3 in 4.38 sec
train epoch: 4, loss: 0.0101, accuracy: 99.69
test epoch: 4, loss: 0.05, accuracy: 98.97
Epoch 4 in 4.44 sec
train epoch: 5, loss: 0.0101, accuracy: 99.69
test epoch: 5, loss: 0.05, accuracy: 99.00
Epoch 5 in 4.47 sec
train epoch: 6, loss: 0.0114, accuracy: 99.69
test epoch: 6, loss: 0.04, accuracy: 99.05
Epoch 6 in 4.48 sec
train epoch: 7, loss: 0.0060, accuracy: 99.82
test epoch: 7, loss: 0.05, accuracy: 99.04
Epoch 7 in 4.46 sec
train epoch: 8, loss: 0.0095, accuracy: 99.74
test epoch: 8, loss: 0.04, accuracy: 98.93
Epoch 8 in 4.42 sec
train epoch: 9, loss: 0.0068, accuracy: 99.81
test epoch: 9, loss: 0.04, accuracy: 99.16
Epoch 9 in 4.39 sec
train epoch: 10, loss: 0.0089, accuracy: 99.76
test epoch: 10, loss: 0.05, accuracy: 99.06
Epoch 10 in 4.48 sec
Total in 44.70 sec
```


JAX/Flaxの実行速度をM2 Mac、CPU on Linux、GPU on Linuxで調べてみた

JAX/Flaxの実行速度 [M2 Mac, CPU, GPU]

結果

.css-12m0k8p{pointer-events:auto;}M2 MacBook Air 13 inch

CPU on Ubuntu

GPU

M2 MacBook Air 13 inch