校赛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

安装成功:

校赛D题:害虫监测:搭建pytorch+yolov5环境_第1张图片

输入:

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:

校赛D题:害虫监测:搭建pytorch+yolov5环境_第2张图片

按照方法换源了也不行,我就直接在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跑通啦,哭哭,我太菜了

 

你可能感兴趣的:(校赛建模,深度学习里面的报错,pytorch,深度学习,人工智能)