红联Linux门户
Linux帮助

Ubutnu16.04安装pytorch

发布时间:2017-07-01 11:02:05来源:linux网站作者:bamtercelboo
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
 
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命令下载安装。
 
6、测试
安装结束之后,打开python输入import torch 若不出现错误就已经安装成功了。
 
7、自己遇到的问题
(1)若心打开一个terminal在python中输入import torch出现不能发现torch moudle的错误,很疑惑,明明pytorch已经安装成功了,为什么还会出现这种错误呢,这个问题应该是环境变量没有配置好的问题,可以自己上网查看,不嫌麻烦的话也可以运行命令 source  ~/.bash_profile  在运行import torch就可以了,但还是要注意安装的过程中就把环境变量配置好。
(2)安装过程中可能需要pip git vim等命令、有时候需要update的,要注意一下。
 
8、参考
Ubutnu16.04安装pytorch
 
本文永久更新地址:http://www.linuxdiyf.com/linux/31815.html