1. Torch C++
1.1 Windows运行
注意libtorch和PyTorch的版本要一致,下载对应版本的方法:
GPU版:cu90表示cuda9,1.0.1表示版本号
https://download.pytorch.org/libtorch/cu90/libtorch-win-shared-with-deps-1.0.1.zip
在下载解压以后,利用CMake-gui生成VS2017工程,CMAKE_BUILD_TYPE要选Release才能正常运行
1.2 Linux运行
和Windows的步骤基本一样,但要注意opencv和gcc版本的问题,libtorch在gcc5以下会遇到std::bad_alloc问题,建议使用GCC5以上,如果遇到imread未定义引用问题,最好用对应版本的gcc编译新的opencv,并在CMakeLists.txt中指定opencv路径
编译如果遇到-ldebug,-loptimized无法找到问题,需要加入第20行
CMakeLists.txt: