파이토치 : 2017년에 공개된 딥러닝 프레임워크
Numpy를 사용한 경우와는 다르게 코드가 간단하다.
GPU를 통한 연산이 가능하다.
CUDA : 엔비디아가 GPU를 통한 연산을 가능하게 만든 API 모델
cuDNN : CUDA를 이용해 딥러닝 연산을 가속해주는 라이브러리
Tensorflow와 Pytorch의 비교
Tensorflow | Pytorch |
---|---|
Define and Run 방식 | Define by Run 방식 |
연산 그래프를 먼저 만들고 실제 연산할 때 값을 전달하여 결과를 얻는 방식 | 그래프를 만듦과 동시에 값이 할당되는 방식 |
직관적으로 받아들이기 어려울 수 있고, 그래프를 정의하는 부분과 이를 돌리는 부분이 분리되므로 전체적으로 코드 길이가 길어지게 된다 | 연산 그래프와 연산을 분리해서 생각할 필요가 없다.연산속도가 더 빠르다 |