Linux下源码编译libtorch

注意：不要编译libtorch1.5.1 ，有内存泄漏问题，第三方库 OpenMP 内存泄漏，而且RReLU也有内存泄漏，从pytorch1.6.0的bug fix中可知。

 

参考官网：https://github.com/pytorch/pytorch/tree/v1.5.1

尝试验证。

安装依赖：

conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi

conda install -c pytorch magma-cu102

安装：

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

没有安装成功，后面补充。

第二次尝试linux下源码编译libtorch，我是在centos7.2下尝试编译的，gcc版本4.8.5.

参考：ubuntu 源码编译libtorch

参考：https://github.com/pytorch/pytorch/blob/master/docs/libtorch.rst

下载源码：

git clone -b master --recurse-submodule https://github.com/pytorch/pytorch.git

进入pyorch文件夹查看历史版本，因为master是最新版本

cd pytorch
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
v1.5.1
v1.5.1-rc1
v1.6.0
v1.6.0-rc1
v1.6.0-rc2
v1.6.0-rc3
v1.6.0-rc4
v1.6.0-rc5
v1.6.0-rc6
v1.6.0-rc7
v1.7.0
v1.7.0-rc1
v1.7.0-rc2
v1.7.0-rc3
v1.7.0-rc4
v1.7.1
v1.7.1-rc1
v1.7.1-rc2
v1.7.1-rc3
v1.8.0-rc1
v1.8.0-rc2
v1.8.0-rc3
v1.8.0-rc4
v1.8.0-rc5

比如我要使用v1.5.1-rc1，执行：

git checkout -b v1.5.1-rc1

安装会遇到两个问题：1、本地需要安装python3。 2、gcc版本4.8.5太低。