cmake -DCMAKE_PREFIX_PATH=/home/XXX/libtorch -DCMAKE_C_COMPILER:FILEPATH=/home/XXX/gcc \
-DCMAKE_CXX_COMPILER:FILEPATH=/home/XXX/g++
