
[Linux] PyTorch 모델을 C++ 포팅으로 사용하기 - ONNX Runtime 설치
·
Linux
라즈베리 파이는 기본적으로 PyTorch의 GPU 가속을 사용할 수 없다. 라즈베리 파이 5의 경우 NPU 가속의 사용 가능성이 있지만, PyTorch에서는 이를 공식적으로 지원하지 않는다. 때문에 경량화 PyTorch를 사용해야 하는데, CPU만으로 동작하기 때문에 라즈베리 파이에서 직접 모델을 학습시키는 것은 매우 느리다. 따라서 모델은 PC에서 GPU 가속으로 학습하고, 이를 .onnx 형식으로 변환해 라즈베리 파이에 업로드한다. 그리고 C++에 포팅하여 데이터 전처리를 하는 방법을 사용한다. 이를 위해서 라즈베리 파이에 ONNX Runtime을 설치해야 한다. 일반적으로는 다음 과정을 거친다.sudo apt update && sudo apt upgrade -ysudo apt updatesudo..