pytorch 源码安装

电脑配置：ubuntu16.04 cuda8.0 cudnn6 anaconda虚拟环境 python2.7

1. 配置依赖 ，设置cuda
   export CMAKE_PREFIX_PATH="$(dirname $(which conda))/…/" # [anaconda root directory]
   conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
   conda install -c mingfeima mkldnn
   conda install -c pytorch magma-cuda80
2. 下载pytorch 源码
   git clone --recursive https://github.com/pytorch/pytorch.git
   cd pytorch
3. 运行安装
   python setup.py install