AI, DataScience
[딥러닝] TensorFlow, Keras, PyTorch 특징 비교, 장단점
이무기뱀술
2022. 2. 15. 10:00
728x90
TensorFlow
- 다양한 언어 지원, 오픈소스
- API: High and Low Level
- 속도: 느림
- 사용성: 불편
Keras
- 파이썬, 오픈소스
- TensorFlow와 함께 실행가능
- 가장 널리 사용
- 보통 간단한 것은 Keras로 구현
- 복잡한 부분은 TensorFlow를 이용할 수 있음
- API: High Level
- 속도: 느림
- 사용성: 매우 편함
PyTorch
- 파이썬, 오픈소스
- 디버깅 가능
- 현업에서 인기 점점 상승
- API: Low Level
- 속도: 빠름
- 사용성: 불편
결론
Keras
- 빠르게 결과물을 보고 싶을 때
- 데이터셋이 작을 때
PyTorch
- 본격적으로 쓰고 싶을 때
- 고성능, 로우레벨API
- 디버깅이 가능함!
- 데이터셋이 클 때
TensorFlow
- 고성능이 필요할 때
- 데이터셋이 클 때
출처:
https://www.simplilearn.com/keras-vs-tensorflow-vs-pytorch-article
728x90