LibTorch 시작하기: C++로 PyTorch 환경설정부터 코드 작성까지

회의 중 나이가 지긋한 직원은 이렇게 묻는다. "그거...C++ 에서도 동작하죠?" 또 지긋지긋한 언어 논쟁이다. Python은 느리고 C++은 빠르다, 등등.. 요즘은 ONNX, TensorRT가 비교적 친절하게 되어 있어서 C++로 추론(Inference)이 가능하다. 다시 말해, 모델은 Python으로 학습 하되 환경에 따라 모델 서빙을 C++로도 할 수 있다는 이야기다. "물론 가능합니다. 모델 학습은 Python으로 만들고..." 대답이 채 끝나기 전에 묻는다. "모델 학습도 C++에서 되게 할 수 없나요? 학습 속도(?)가 더 빠르잖아요?" 정말, 여기서 한숨이 나오는 부분이다. 모든 걸 C++로 바꾸는건 나의 영역이 아니라는 것을 설명하는 것도 지친다. 가능하기야 하겠지..ㅠ 이번에는 PyTorch의 C++ 버전, LibTorch 설치 및 환경 설정에 대해 작성해본다. 1. GPU 연산을 사용할 것이라면 CUDA, cuDNN을 설치한다. 본인의 그래픽카드에 맞게 설치해야...

원문링크 : LibTorch 시작하기: C++로 PyTorch 환경설정부터 코드 작성까지