Ubutnu16.04安装pytorch

 windows 安装参考 https://bamtercelboo.github.io/2017/07/26/ubuntu-windows%E5%AE%89%E8%A3%85pytorch/

1、下载Anaconda3

  首先需要去Anaconda官网下载最新版本Anaconda3(https://www.continuum.io/downloads),我下载是是带有python3.6的Anaconda3-4.4.0-Linux-x86_64.sh。

2、安装Annconda3

   bash Anaconda3-4.4.0-Linux-x86_64.sh   在home/ubuntu出现anaconda3文件夹(注:ubuntu是系统用户名、下同)。

   source ~/.bashrc

   安装的过程中需要选择三个yes、一个也不能少,这样就不用配置环境变量了

3、配置环境变量

   gedit ~/.bashrc  加入下面那行命令

   export CMAKE_PREFIX_PATH=/home/ubuntu/anaconda3/bin

   gedit ~/.bash_profile   加入下面那行命令
   export PATH="/home/ubuntu/anaconda3/bin:$PATH"

4、还需要安装一些依赖库

   conda install numpy mkl setuptools cmake gcc cffi 
   conda install -c soumith magma-cuda75     # or magma-cuda80 if CUDA 8.0
   (若出现conda not found之类的错误,运行命令source ~/.bashrc 、然后在执行)

5、源码安装pytorch

   (1)可以用github上的这个链接(https://github.com/pytorch/pytorch)的源码,git cone(首先要建立git与github的ssh)到本地,运行
     pip install -r requirements.txt 、python setup.py install安装即可。
     注:不推荐使用这个github上面的源码安装、因为可能是因为pytorch版本的问题,导致我后面使用Tensor张量的时候出现了一下问题,例如经过  Temsor.max()函数之后把我的二维的变成了一维。
   (2)http://pytorch.org/ ,conda命令下载安装。
   (3)conda install -c peterjc123 pytorch=0.1.12     运行这个命令就OK
    pytorch最近官网源码有了新的更新,但是Conda命令安装的可能还是以前没有更新的,更新之后源码问题应该也改善了,所以也可以选择源码安装。

6、测试

   安装结束之后,打开python输入import torch 若不出现错误就已经安装成功了。

7、自己遇到的问题

  (1)若心打开一个terminal在python中输入import torch出现不能发现torch moudle的错误,很疑惑,明明pytorch已经安装成功了,为什么还会出现这种错误呢,这个问题应该是环境变量没有配置好的问题,可以自己上网查看,不嫌麻烦的话也可以运行命令 source  ~/.bash_profile  在运行import torch就可以了,但还是要注意安装的过程中就把环境变量配置好。
  (2)安装过程中可能需要pip git vim等命令、有时候需要update的,要注意一下。

8、参考

http://blog.csdn.net/u013066730/article/details/62217401

    原文作者:pytorch
    原文地址: https://www.cnblogs.com/bamtercelboo/p/7074366.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞