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来运算。