pytorch安装

1,去官网选择和自己OS相匹配的pytorch版本，生成下面命令。（注意：不能生成时需要fan qiang哟）

sudo pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl

pip install torchvision

2,测试是否安装成功

进入python

import torch

3,测试是否支持nvidia的显卡并支持cuda

torch.cuda.is_available()  #看看是否支持cuda

x = x.cuda()

y = y.cuda()

x+y           #这里的x和y都是tensor，使用cuda函数以后，x和y的所有运算均会调用gpu来运算。