【环境搭建】在Ubuntu16.04上编译安装PyTorch（caffe2已经并入PyTorch）

编译安装PyTorch

# --recursive意在clone子项目，在PyTorch源码中，third_party文件夹下包含子项目
# 如果仅git clone，clone下来的源码会缺少子项目中应有的文件
git clone --recursive https://github.com/pytorch/pytorch.git

# 选择pytorch版本
# 不同版本关键字可在github源码页Branch中查看得到
cd pytorch
git checkout v1.2.0

sudo apt-get install libgtest-dev

sudo apt-get install libiomp-dev

sudo apt-get install libopenmpi-dev

sudo apt-get install openmpi-bin

sudo apt-get install openmpi-doc

【环境搭建】gflags和glog的安装
【环境搭建】在ubuntu16.04上编译安装ffmpeg和opencv
【环境搭建】linux上编译安装caffe框架+Makefile.config文件详解
上面三篇博文中部分apt安装的依赖包这里也需要，不再赘述。
在PyTorch源码中的requirements.txt文件列出的包可以选择用pip安装

# 到这里PyTorch编译安装完成
sudo python3 setup.py install

遇到的问题

1. 在执行pip install future时报错如下：

“ImportError: ‘module’ object has no attribute ‘check_specifier’”

原因是setuptools版本低，升级至最新版即可
安装最新版本setuptools的过程在【环境搭建】linux上pip换源中有说明

# 查看setuptools版本
easy_install --version

2. 注意不要在PyTorch源码目录下用python import torch，否则会报错No module named _C.

pip安装PyTorch

https://pytorch.org/

测试是否安装成功

在python中：

import torch
import torchvision
torch.cuda.is_available()
torch.version.cuda
torch.__version__

其他

caffe2的官方手册中安装caffe2步骤如下：

git clone https://github.com/pytorch/pytorch.git && cd pytorch
# 这里这句命令和 git clone --recursive功能一致
# 即将子项目内容下载下来
git submodule update --init --recursive
python setup.py install

结语

如果您有修改意见或问题，欢迎留言或者通过邮箱和我联系。
如果我的文章对您有帮助，转载请注明出处。