9.Ubuntu 16.04安装pytorch教程

一、PyTorch官网安装命令合集

        pytorch 有中文网站，可以直接去找命令合集。PyTorch中文网站

        1.先查看 CUDA 版本：

cat /usr/local/cuda/version.txt

        2.查看 CUDNN 版本：

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

 二、使用conda安装pytorch(方式一)

        如果是anaconda|python用户，进入pytorch官网(https://pytorch.org/)，如下页面选择适合自己的配置进行安装。其中pytorch build选择stable(1.0)，安装时间略长，不知道是因为网速原因还是没有换镜像，差不多安装了一下午~~~

         执行如下命令即可(如果第一次报错就多执行两次)：

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

         如果执行多次还是报错，那么考虑下面的方法......

        修改conda镜像源

        在终端运行以下命令修改镜像源，可以明显加速安装。

#优先使用清华conda镜像
conda config --prepend channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
#据说加上下面清华大学conda的main镜像，可以安装、更新最新版本的软件，没试过~~~
#conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
#也可以选用科大conda镜像
conda config --prepend channels http://mirrors.ustc.edu.cn/anaconda/pkgs/free/

        要查看镜像源是否安装成功的话，建议终端中运行以下命令：

conda config --set show_channel_urls yes

        会生成一个~/.condarc文件，运行cat命令查看文件内容

cat ~/.condarc

        显示内容为

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

         如果要切回默认源，则执行

conda config –remove-key channels

安装pytorch

        在终端输入如下命令，安装pytorch(感觉并没有加快安装速度，也用了将近一下午时间，只是安装成功率提高了)：

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

         测试

        在终端中按以下顺序运行命令:

python

        在打开的python命令行中运行如下命令，不报错说明安装成功~~~

import torch
import torchvision
print(torch.__version__)

         注：其实到此pytorch就算安装完成了，但pytorch版本过低的话import torch.optim.lr_scheduler as lrs会报错，好像pytorch0.2以后才有的lr_scheduler方法，踩过的坑特意注明一下。

三、使用pip或pip3安装pytorch 0.3.1和torchvision(方式二)

        除了使用conda以外，我们还可以使用pip管理工具，当我们python2.x时，使用pip安装，python3.x时候，可以使用pip3安装(但安装了python3.x默认就是python3，而且python3可以使用pip安装，这个还不太清楚怎么回事儿~~~)：

包管理器	服务器	CUDA	Python版本	操作方法