libtorch指定版本在linux系统下的源码编译

一. 环境准备

1. centos7或者ubuntu
2. miniconda安装，请参考Linux下conda安装及使用
3. cuda-10.2
4. cudnn7.6.5
5. pytorch v1.2.0

---

二. 依赖库安装（激活conda环境,例如: conda activate libtorch120）
1.Common

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

2.Add LAPACK support for the GPU if needed

conda install -c pytorch magma-cuda102 # or [magma-cuda92 | magma-cuda100 ] depending on your cuda version

3.clone source code

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch

#switch branch
git tag 
git checkout  v1.2.0

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

注意：以上操作可以在其他环境下执行完，再拷贝到目标环境下

---

三.编译
1.设置环境变量

#如果gcc<= 4.8,否则装有更高版本的gcc,这一步可以跳过
export LD_LIBRARY_PATH=/XX/anaconda3/envs/libtorch120/lib:$LD_LIBRARY_PATH

2.build

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

3.待编译完拷贝出pytorch/torch/lib，即可。