ubuntu16.04详细安装pytorch(GPU)

安装pytorch要安装两个模块：torch和torchvision。torch是主模块，用来搭建神经网络。torchvision是辅模块，里面有搭建好的网络可以直接用。

1.安装pip3，ubuntu自带python3.5和2.7,所以没装python了

sudo apt-get install python3-pip

2.install numpy, scipy

pip3 install numpy

pip3 install --upgrade pip
pip3 install scipy
3.下载Anaconda3-5.0.1-Linux-x86_64.sh再安装
 bash Anaconda3-5.0.1-Linux-x86_64.sh
4.设置环境变量(a320是系统用户名)
gedit ~/.bashrc   

加入下面这行：

export PATH="/home/a320/anaconda3/bin:$PATH"

source ~/.bashrc

5.安装依赖库

conda install numpy mkl setuptools cmake gcc cffi

conda install -c soumith magma-cuda80（我用的是cuda8.0）
6.下载pytorch并安装torch

conda http://pytorch.org

conda install -c peterjc123 pytorch=0.1.12

7.最后安装torchvision模块

conda install pytorch torchvision -c soumith

8.检测有没有装好pytorch

在终端输入python

import torch

import torchvision

没有错误就安装好了pytorch。