Linux安装pytorch的具体过程以及其中出现问题的解决办法

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)
原文地址:https://www.cnblogs.com/xueqiuqiu/p/7505966.html