之前折腾了yolov3
现在直接上yolov5,话不多说。
(1)安装cuda
首先打开官网
https://pytorch.org/get-started/locally/
我们可以看到需要的是cuda11
直接
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
查看一下版本直接yes
安装完成
(2)测试
发现问题
解决:版本问题
不得不说还是这个老哥的版本强呀!