校赛D题:害虫监测:搭建pytorch+yolov5环境

用的这个yolov5模型:

yolov5-master(cow).rar_免费高速下载|百度网盘-分享无限制 (baidu.com)

提取码:2022

电脑须有NVIDIA独立显卡,(绝大多数电脑都符合pytorch要求,不用管CUDA ,CUDNN

假设你已经安装了Anaconda

anaconda prompt打开

输入:

conda create -n Pytorch_GPU python=3.8

安装成功:

输入:

activate Pytorch_GPU

输入:

conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2

最好安装 1.8.0 的Pytorch版本,电脑都能满足这个版本,否则YOLOv5模型对高版本的torch会报错,我一开始是1.11.2版本的torch,直接去闯就报错了,一查就是版本太高的问题

报错:PackagesNotFoundError: The following packages are not available from current channels:

按照方法换源了也不行,我就直接在pycharm终端安装torch==1.8.0,在官网上下载的 

10分钟安装pytorch1.8_Alex抱着爆米花的博客-CSDN博客_pytorch1.8

结果在方法二里面先装的torchversion,torch就下不来了,用豆瓣源,发现pycharm终端-i命令可以换源

 pip install "torchvision-0.9.0+cpu-cp39-cp39-win_amd64.whl"-i http://pypi.douban.com/simple --trus
ted-host=pypi.douban.com

发现torch安装好了

如果还是红色的就再重启一下pycharm试试 

opencv-python

opencv-contrib-python

 pip install -i https://pypi.douban.com/simple --trusted-host pypi.douban.com opencv-python



pip install -i https://pypi.douban.com/simple --trusted-host pypi.douban.com opencv-contrib-python

尝试一下yolov5能跑了不,在pycharm终端里面输入:

python detect.py --source data/images/bus.jpg --weights pretrained/yolov5s.pt

出现哪个包没安装就再用上面那个源或者再搜一下国内其他源的名字去下好所有包

最后再来一次上面那个命令

得到结果!!!!

看到下面那个results saved to 就是结果保存目录

 终于把yolo跑通啦,哭哭,我太菜了

 

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值