目标检测:yolov5环境搭建,3分钟弄懂,避开环境不适合的坑

解决问题

yolov5环境搭建半天,还是到处是bug。所以我对之前的坑找了一条比较简单而有效的路。

准备:
1、conda
2、yolov5的源码
https://github.com/ultralytics/yolov5

操作步骤

第一步:创建一个python环境

conda create -n yolov5 python=3.7

👉:
1、-n 后面写自己环境的名字
2、python建议用3.7 太高容易帮官方踩坑,太低可能少功能

conda activate yolov5 #进入yolov5这个环境

第二步:安装torch、torchvision等

查看自己的cuda版本(因为要和torch适配)
大家可以通过终端查看。

nvidia-smi

nvcc --version

在这里插入图片描述
在这里插入图片描述
👉:
1、这里就有疑问,为什么cuda版本不同,这里不管,只用看nvcc --version的。所以我这里是10.2版本。

然后在torch官网查看cuda匹配的版本,并复制相关安装指令
yolov5官方要求torch至少是1.7的torch,我选个1.8.1
在这里插入图片描述
所以我这里安装torch的指令如下

pip install torch1.8.1+cu102 torchvision0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
然后我这里numpy都给我配套安装了。。。

第三步:安装yolov5其它库

进入下载好的yolov5源码
在这里插入图片描述
把上一步安装过的包,比如torch等注释掉,不要覆盖第二步搭建好的torch包
在这里插入图片描述
最后在该文件下输入

pip install -r requirements.txt

在这里插入图片描述
训练和测试后面有空再说。

  • 2
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值