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