linux下安装GPU 版本 torch torchvision等最方便的方式

其实很简单而且很快，就是安装YOLOV5中的requirement.txt 中的内容即可，即去这个地方
点我进入YOLOV5
找到requirement.txt 将其中的内容复制下来，放到你的（随便起名最好还是requirement.txt）.txt中,然后在当前目录下打开一个终端，然后执行 pip install -r requirements.txt或者是你起的名字，则执行pip install -r 你起的名字.txt 坐等安装完即可，这种方式的好处是他会根据你系统选择正确的方式安装

上边的还是不能用：用下边这个即可，即更换安装源，这样速度会快些

conda install pytorch torchvision torchaudio cudatoolkit -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

注：安装之前先将之前安装的pytorch卸载了
下边是验证：

a=torch.Tensor([1,2])
a=a.cuda("cuda:0")
print(a)

效果如下：