컴퓨터 공부/데이터사이언스
텐서플로우(Tensorflow)와 파이토치(PyTorch)
구깨비
2021. 12. 27. 10:24
1. 딥러닝 프레임워크
- 딥러닝 구현을 쉽고 편하게 할 수 있도록 도와주는 소프트웨어
- 종류: TensorFlow, PyTorch, Keras, Caffe / Caffe2, Theano, MXNET, CNTK 등...
2. TensorFlow
- Python을 사용하는 오픈소스 소프트웨어 라이브러리
- Originally developed by Google Brain Team
- DeepMind, Uber, AirBnB, Dropbox 같은 기업에서 사용
3. PyTorch
- Lua라는 언어로 작성된 Torch library의 Python 후속 제품
- Facebook AI Research group에서 개발됨
- Twitter, Salseforce, Stanford University, UDACITY 등에서 사용
4. TensorFlow vs PyTorch
TensorFlow | PyTorch |
Static -Define and Run | Dynamic - Define by Run |
Debugging이 어려움 | Debugging이 쉬움 |
Tensorboard라는 강력한 시각화 툴 보유 | 시각화 툴 부족 |
Rich community | Growing Community |
상업용, 연구용으로 많이 사용 | 연구용으로 많이 사용 |
모바일용 라이브러리를 따로보유, TensorRT같은 최적화 tool보유 |
모바일용 라이브러리 따로 없음, 최적화 tool없음 |
진입장벽 상대적으로 높음 | 진입장벽이 상대적으로 낮음 |