其实很简单而且很快,就是安装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)
效果如下: