본문 바로가기
컴퓨터 공부/데이터사이언스

텐서플로우(Tensorflow)와 파이토치(PyTorch)

by 구깨비 2021. 12. 27.

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없음
진입장벽 상대적으로 높음  진입장벽이 상대적으로 낮음

댓글