1.安装Anaconda
安装步骤参考了官网的说明:https://docs.anaconda.com/anaconda/install/linux.html
具体步骤如下:
首先,在官网下载地址 https://www.anaconda.com/download/下载linux版本,这里选用python 3.6版本的anaconda.
然后, 将下载好的Anaconda3-4.4.0-Linux-x86_64.sh放到/usr/tiny目录下,并进入该目录
在当前目录下用bash命令安装anaconda
root bash Anaconda3-4.4.0-Linux-x86_64.sh
安装完毕后,输入如下命令使得配置生效
source ~/.bashrc
为验证安装正确性,可以通过如下命令进行测试
conda list
2.用conda安装/更新pytorch的依赖库
首先,激活Anaconda配置文件
source ~/.bashrc
使用conda install命令可以方便快捷得安装或者更新所需的内容
conda install numpy pyyaml mkl setuptools cmake gcc cffi
3.安装pytorch
本文对pytorch官网给出的三种安装方式分别进行了测试,结果如下:
首先,使用conda进行安装,命令如下,其中cuda80可以根据cuba版本调整为cuda75或者直接省略
conda install pytorch torchvision cuda80 -c soumith
因为下载速度只有几KB,忍无可忍直接终止了进程。
然后直接从官网下载了source文件
git clone --recursive https://github.com/pytorch/pytorch
然后用setup.py进行安装
python setup.py install
安装过程显示报错,缺失python.h文件,于是把python-dev和python3-dev一起装了一遍
sudo apt-get install python-dev
sudo apt-get install python3-dev
再次运行
python setup.py install
显示把torch安装到了python2.7的库文件夹里,本来想删掉python-dev再安装一遍pytorch,没有找到相应的办法,遂考虑通过pip进行安装
通过pip进行pytorch安装的过程如下:
pip install http://download.pytorch.org/whl/cu75/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl pip install torchvision
下载速度在几MB左右,安装过程十分迅速,且没有报错。
安装完毕之后,激活Anaconda配置文件
source ~/.bashrc
进入python命令行模式:
python
测试安装是否正确:
from __future__ import print_function import torch a = torch.ones(5) print(a)