ubuntu 源碼編譯libtorch


pytorch/libtorch qq群: 1041467052

有一點兒感悟就是:
一定要去官網找一手資料,百度出來的都是個人根據官網來的,這個就隨便看看了。
官網:https://github.com/pytorch/pytorch
首先我跑的是pytorch 1.3版本的,conda安裝好的,現在需要源碼編譯。按照官網的流程,需要先安裝依賴包。
我是切換到pytorch1.3的conda虛擬環境進行的:這里是官網的教程:

其中,git clone --recursive https://github.com/pytorch/pytorch 是下載的最新版本,而我需要1.3的。
下載完成后:

git tag

會出來:

v0.1.1
v0.1.10
v0.1.11
v0.1.12
v0.1.2
v0.1.3
v0.1.4
v0.1.5
v0.1.6
v0.1.7
v0.1.8
v0.1.9
v0.2.0
v0.3.0
v0.3.1
v0.4.0
v0.4.1
v1.0.0
v1.0.0a0
v1.0.1
v1.0rc0
v1.0rc1
v1.1.0
v1.1.0a0
v1.2.0
v1.2.0a0
v1.3.0
v1.3.0a0
v1.3.1
v1.4.0
v1.4.0a0
v1.4.1
v1.5.0
v1.5.0-rc1
v1.5.0-rc2
v1.5.0-rc3
v1.5.0-rc4
v1.5.0-rc5

然后敲:

git checkout v1.3.1

然后再:

git submodule sync
git submodule update --init --recursive


export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
python setup.py install

然后是漫長的等待
好久之后,沒有報錯,然后我不知道我需要的庫生成在哪里,然后往上翻日志:

- ******** Summary ********
--   CMake version         : 3.6.3
--   CMake command         : /usr/local/bin/cmake
--   System                : Linux
--   C++ compiler          : /usr/bin/c++
--   C++ compiler version  : 5.4.0
--   CXX flags             :  -fvisibility-inlines-hidden -fopenmp -Wnon-virtual-dtor
--   Build type            : Release
--   Compile definitions   : ONNX_ML=1
--   CMAKE_PREFIX_PATH     : /data_1/Yang/software_install/Anaconda1105/bin/../;/usr/local/cuda
--   CMAKE_INSTALL_PREFIX  : /data_2/everyday/0429/pytorch/torch
--   CMAKE_MODULE_PATH     : /data_2/everyday/0429/pytorch/cmake/Modules;/data_2/everyday/0429/pytorch/cmake/public/../Modules_CUDA_fix
-- 
--   ONNX version          : 1.4.1
--   ONNX NAMESPACE        : onnx_torch
--   ONNX_BUILD_TESTS      : OFF
--   ONNX_BUILD_BENCHMARKS : OFF
--   ONNX_USE_LITE_PROTO   : OFF
--   ONNXIFI_DUMMY_BACKEND : OFF
-- 
--   Protobuf compiler     : 
--   Protobuf includes     : 
--   Protobuf libraries    : 
--   BUILD_ONNX_PYTHON     : OFF
-- Found CUDA with FP16 support, compiling with torch.cuda.HalfTensor
-- Removing -DNDEBUG from compile flags
-- MAGMA not found. Compiling without MAGMA support
-- Could not find hardware support for NEON on this machine.

找到
-- CMAKE_INSTALL_PREFIX : /data_2/everyday/0429/pytorch/torch
恩,安裝到了目錄torch下面了。
然后用測試代碼,cmakelist,可以加載模型。應該是可以了。

還有注意就是cuda這些環境需要配置好。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM