安装pytorch 及torchvision

根据cuda及Python版本下载对应的pytorch下载地址，有时进不去多刷新几遍

ctrl+f 搜索需要的torch ，cu+cuda版本号（cuda11.0就搜cu110）再根据Python版本下载对应的torch

再搜索cu110/torchvision下载对应的torchvision
切换到文件下载的路径，然后指定Python版本安装torch及torchvision

//根据自己使用的Python版本安装， --user不需要改动
$ python3.8 -m pip install --user torch torch-1.7.0%2Bcu110-cp38-cp38-linux_x86_64.whl   
$ python3.8 -m pip install --user torchvision-0.8.0-cp38-cp38-linux_x86_64.whl

安装完成后使用以下检测是否能使用

import torch
a = torch.Tensor(5,3)
a=a.cuda()
print(a)