在GTX3060上跑yolov5

针对3060显卡与CUDA版本不匹配的问题,博客建议通过从PyTorch官网下载更高版本的稳定版PyTorch来覆盖Yolov5自带的CUDA10.2,以便利用3060进行模型训练。对于外部CUDA的使用,作者表示将进一步研究。
摘要由CSDN通过智能技术生成

现在你想要深度学习入门就需要先安装cuda等环境,网上也有很多相关的安装教程,但是我们一般都是从yolov5入手的,根据上面的requestment下载需要的环境,但是它下载的是自带cuda10.2的pytorch,这就导致你从外部下载的cuda用不上,或者说换不了版本,而且我现在也不知道怎么用外部的cuda,网上都是说3060不适应cuda10.2的版本,要下载10以上的,但是你如果是yolov5跑的,那即使你下了也没用,yolov5会自动使用那个他下载的版本,也就是你用不了你下载的高版本

所以我们需要从pytorch官网下载稳定版本的更高级的pytorch去覆盖yolov5下载的,至于外部的cuda怎么用就等之后慢慢研究吧。
在这里插入图片描述这样你就可以开始用你的3060训练模型啦。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值