之前折腾了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)测试
发现问题

解决:版本问题
不得不说还是这个老哥的版本强呀!

最低0.47元/天 解锁文章
4111

被折叠的 条评论
为什么被折叠?



