완벽하게 동일한 세팅(CPU/RAM/storage)에서 비교한 것이 아니므로
오로지 GPU의 성능을 비교했다고 하긴 어려움
적당한 비교(just curious)로 받아들이면 좋을듯
동일한 학습 모델, 데이터 셋에서의 학습 성능 비교
- 첫 2 epoch에 걸린 시간을 비교
Framework: Pytorch
Model: efficientnet lite0
Batch size: 20
Image size: 384x384
데이터 수: 약 3만장
시스템:
결과:
결론
1. 역시 Nvidia가 좋다.
2. 학습속도는 조금 느릴지라도 AMD의 높은 GPU RAM은 무거운 모델 학습/구동에 좋을 수도 있겠다.
'컴퓨터 > 머신러닝 (Machine Learning)' 카테고리의 다른 글
| Super resolution 모델, HAT, inference 사용 정리 (0) | 2022.12.19 |
|---|---|
| TensorRT Docker 사용 정리 (0) | 2022.12.13 |
| Ubuntu, ROCm, AMD GPU, Docker, Pytorch 환경에서 딥러닝 정리 (0) | 2022.12.10 |
| Pytorch, grad-cam 사용 정리 (0) | 2022.12.10 |
| Ubuntu, TensorRT로 Yolov5 inference (0) | 2022.12.05 |
