PyTorch 学习之Ubuntu16.04 下安装 PyTorch（conda与pip两种方式安装）

一、pip安装Pytorch

     这里已经默认已经安装好了显卡驱动，cuda等。我这里安装的是cuda9与cudnn7，显卡驱动号：Driver Version: 384.130 ；

   1、进入PyTorch官网：

           pytorch 官网

进去后，点击左下角 previous versions of PyTorch

找到自己所需要下載的版本（torch 与torchvison）.如系統,cuda版本,python版本。

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

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

torch-0.3.1-cp35-cp35m-linux_x86_64.whl爲你下載的文件。
安装完PyTorch后，要安装torchvision，方法如下：

目前我了解的torc与torchvision版本的匹配：

  torch  1.1   torchvision0.3

  torch  1.2   torchvision0.4

  torch 1.0一下的  torch0.2或者0.1

依然在下载torch那的链接下载torchvision，同样的安装方式；

sudo pip3 install torchvision-版本号.whl

卸载：

sudo pip3 uninstall torchvision

二、conda安裝Pytroch

按照正常安裝，是下圖所示，直接使用命令進行安裝；

但是，安装pytorch官方指定的方法conda下载torch太慢。因此交給大家一个方法，

首先添加一個清華源來加快速度；

解决步骤：

第一步：进入添加清華源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

然後添加pytroch源：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

最後安裝：

conda install pytorch torchvision cudatoolkit=9.0（这个时候要注意，我们已经吧-c pytorch给删除了）

 解释一下，就是我们没有使用pytorch的源，而改成清华镜像的源了；

三、檢驗是否安裝成功

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

python3

进入python3的环境

然后输入：

import torch

import torchvision

測試cuda是否可用：

import torch
print(torch.cuda.is_available())

如果没有报错，就证明安装成功！！如果還有問題，請在下面留言