注意:不要编译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.
参考: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太低。