PyTorch源码安装小记

Torch是Facebook AIR去年6月开源的深度学习框架，向G家的Tensorflow看起，基于Lua开发，可以充分利用GPU进行计算加速。PyTorch是FAIR今年1月发布的，在原来基础上提供python接口支持？也是正面刚支持py的tf吧。

本来安装PyTorch比较简单，按照官网，装完anaconda后一句conda命令就可以安装：

然而最后还是倒在墙下。

没有办法，只能选择源码编译安装了：https://github.com/pytorch/pytorch

对于Ubuntu 14.04，把pytorch源码clone到本地后：

export CMAKE_PREFIX_PATH=[anaconda root directory]

# Install basic dependencies
conda install numpy pyyaml mkl setuptools cmake gcc cffi

# Add LAPACK support for the GPU
conda install -c soumith magma-cuda80 # or magma-cuda75 if CUDA 7.5

python setup.py install

编译完后急切想知道是否安装成功，于是熟练地import torch，然后就会报错：

ImportError: No module named _C 

查了下pytorch repo的issues，果然很多人遇到：https://github.com/pytorch/pytorch/issues/7

作者淡定说，不要在pytorch项目根目录下导入torch……

然后就可以了。

下一步就基于PyTorch跑个小Demo吧。