首页 > 代码库 > Ubutnu16.04安装pytorch
Ubutnu16.04安装pytorch
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、参考
http://blog.csdn.net/u013066730/article/details/62217401
Ubutnu16.04安装pytorch
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。