Ubutnu16.04安装pytorch

时间：2017-06-24 21:40:29 阅读：553 评论：0 收藏：0 [点我收藏+]

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

原文：http://www.cnblogs.com/bamtercelboo/p/7074366.html

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年09月23日 (328)
2021年09月24日 (313)
2021年09月17日 (191)
2021年09月15日 (369)
2021年09月16日 (411)
2021年09月13日 (439)
2021年09月11日 (398)
2021年09月12日 (393)
2021年09月10日 (160)
2021年09月08日 (222)