PyTorch 深度学习【一】Ubuntu16.04 下安装 PyTorch

进入PyTorch官网： 
PyTorch

在PyTorch主页上右下角点击“Click here for previous versions of PyTorch”，如下图：

然后在弹出的界面，根据你自身装的CUDA版本去选择对应的whl文件进行下载：

要用GPU的话，就要选择含cu的版本，我的电脑安装的是CUDA 8，然后我想安装在系统的Python3.5上，所以选择如下：

其中cp35代表python3.5，同理，cp27就是python2.7

前面的0.4.0或者0.3.1就是PyTorch的版本数，建议使用0.3.1，因为现在很多开源的代码都是基于0.3的，不同版本的函数有些不一样

 

进入PyTorch的下载目录，使用pip3命令安装：（python3是用pip3，python2就直接用pip，根据自己的情况，注意这点不同）

sudo pip3 install torch-0.3.1-cp35-cp35m-linux_x86_64.whl

安装完PyTorch后，要安装torchvision，方法如下：

sudo pip3 install torchvision

 

安装成功PyTorch和torchvision后，在终端输入：

python3

进入python3的环境

然后输入：

import torch
import torchvision

如果没有报错，就证明安装成功！！

 

 

如果出现下面的报错：

ImportError: numpy.core.multiarray failed to import

说明你的numpy版本需要更新，直接用pip3去更新numpy：

sudo pip3 install numpy

 

最后我们再验证下GPU能否使用：

print(torch.cuda.is_available())

返回True，表明已经可以使用GPU了！

 

成功安装显示如下：

 

 

补充：

1.查看当前PyTorch版本：

import torch
print(torch.__version__)

2.卸载PyTorch：

sudo pip3 uninstall torch

3.直接安装指定版本的PyTorch

sudo pip3 install pytorch==0.3.1