一. 环境准备
- centos7或者ubuntu
- miniconda安装，请参考Linux下conda安装及使用
- cuda-10.2
- cudnn7.6.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，即可。
