TorchDemo
TorchDemo copied to clipboard
Pytorch libtorch demo
Torch Demo
Pytorch LibTorch demo
setup
- Download C++ LibTorch library at https://pytorch.org/get-started/locally/
- Download and build C++ library OpenCV at https://opencv.org/releases/
- Setup your libs path on CMakeList.txt
set(Torch_DIR ~/your/path/here/libtorch/share/cmake/Torch)
set(OpenCV_DIR ~/your/path/here/opencv-4.7.0/build)
Hints
For libtroch "Pre-cxx11 ABI" can cause compiler conflicts, instead use cxx11 ABI
Step
- python3 tracing.py
- build the c++ program
- ./TorchDemo ../model.pt ../dog.png ../synset_words.txt
more informattion
https://jackan.cn/2018/12/23/libtorch-test/#more