1.安装cuda
1.1 查看ubuntu的英伟达显卡驱动
nvidia-smi
得到
驱动版本是384.130，比较老，所以需要下载旧版本的cuda
1.2 查看显卡是否支持CUDA计算
然后去到这里https://developer.nvidia.com/cuda-gpus查看你的显卡是否在表中，在的话你显卡就是支持CUDA计算的（CUDA-capable）。
结果是支持: 可以继续
1.3 安装cuda
根据英伟达cuda手册查看该驱动版本支持的cuda版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-new-features
通过上表可得到支持安装cuda 8.0，安装如下
$ conda install cudatoolkit=8.0
$ conda install cudnn
安装完毕后，通过如下命令查看cuda版本信息
$ conda list cudatoolkit
# Name Version Build Channel
cudatoolkit 8.0 3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
2.安装pytorch
1.打开pytorch官网
https://pytorch.org/get-started/locally/
这里只有最新的cuda对应的pytorch，如果使用老版本的cuda，那么需要打开如下:
v1.0.0之后的版本支持cuda8
# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
# <font size=4><font color=000099>1.安装cuda</font></font>
## <font size=3><font color=009999>1.1 查看ubuntu的英伟达显卡驱动</font></font>
```c
nvidia-smi
得到
驱动版本是384.130，比较老，所以需要下载旧版本的cuda
1.2 查看显卡是否支持CUDA计算
然后去到这里https://developer.nvidia.com/cuda-gpus查看你的显卡是否在表中，在的话你显卡就是支持CUDA计算的（CUDA-capable）。
结果是支持: 可以继续
1.3 安装cuda
根据英伟达cuda手册查看该驱动版本支持的cuda版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-new-features
通过上表可得到支持安装cuda 8.0，安装如下
$ conda install cudatoolkit=8.0
$ conda install cudnn
安装完毕后，通过如下命令查看cuda版本信息
$ conda list cudatoolkit
# Name Version Build Channel
cudatoolkit 8.0 3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
2.安装pytorch
1.打开pytorch官网
https://pytorch.org/get-started/locally/
这里只有最新的cuda对应的pytorch，如果使用老版本的cuda，那么需要打开如下:
v1.0.0之后的版本支持cuda8
# CUDA 8.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda80 -c pytorch
# 失败，采用pip安装
pip install http://download.pytorch.org/whl/cu80/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
cuda安装参考
https://blog.csdn.net/xiangduixuexi/article/details/105643254
pytorch安装参考
https://blog.csdn.net/u014546828/article/details/108077306