pytorch 自去年年底诞生以来日趋火爆。PyTorch 是 Torch 在 Python 上的衍生. 因为 Torch 是一个使用 Lua 语言的神经网络库, Torch 很好用, 但是 Lua 又不是特别流行, 所有开发团队将 Lua 的 Torch 移植到了更流行的语言 Python 上。下面主要讲讲在 centos 7 下的环境搭建。
1. 安装 python35
centos 7 自带的 python 是 2.7 版本,然而博主习惯 python3.x,于是现安装 python35.浏览器打开 https://www.python.org/ftp/python/ 查看最新的Python版本,标记为3.A.B,此处为3.5.4。
wget https://www.python.org/ftp/python/3.5.4/Python-3.5.4.tgz
tar -zxvf Python-3.5.4.tgz
cd Python-3.5.4
./configure
make
make install
python3 -V //查看是否安装成功
2. 安装 pip3
wget --no-check-certificate https://github.com/pypa/pip/archive/9.0.1.tar.gz
tar -zvxf 9.0.1.tar.gz
cd pip-9.0.1
python3 setup.py install
pip3 -V //查看是否安装成功
3. 安装 pytorch,torchvision
pytorch 的官网提供了自定义组合下载命令,但是由于天朝有墙看不到安装命令,于是尝试源码安装:
git clone git@github.com:pytorch/pytorch.git
cd pytorch
python3 setup.py install
git clone git@github.com:pytorch/vision.git
cd vision
python3 setup.py install
对 torchvision 的安装也可直接使用 pip3:
pip3 install torchvision
此时,不出意外,已经安装成功了,终端输入 python3 ,查看 torch 是否导入成功(也可使用 pip3 list 查看):
python3
import torch
import torchvision