Windows下 YOLOv5配置、部署、及遇到的问题

YOLOv5配置

Yolov5前期配置参考该博客

简单快速建立pytorch环境YOLOv5目标检测 模型跑起来(超简单)_Big data达闻西的博客-CSDN博客_pytorch yolo

本人先期已经配置完成pytorch aconda 未配置的童鞋见以下博客

安装PyTorch详细过程_MC云鸷骚峰的博客-CSDN博客_pytorch安装

使用conda做虚拟环境管理之前建议先下好YOLOv5的包 把虚拟环境建在包的文件夹里 之后再在里面下载pytorch

之后检查配置

(5条消息) 检查pytorch是否安装成功、查看torch和cuda的版本_KyrieLiu52的博客-CSDN博客_怎么验证pytorch装好了

问题

1.无法创建虚拟空间

解决CommandNotFoundError: Your shell has not been properly configured to use ‘conda activate’_ironartisan的博客-CSDN博客

先conda init cmd.exe(在window下用命令行的shell是这个)

之后出现InvalidArchiveError

Conda报错InvalidArchiveError_牧阳MuYoung的博客-CSDN博客_invalidarchiveerror

修改conda文件夹权限

问题解决

2.下载报错

给pip跟换国内镜像源

Windows命令行更换pip国内源_mcg@xisu的博客-CSDN博客_pip换源命令

conda换国内镜像源

Anaconda 换国内源_scl52tg的博客-CSDN博客_conda 换源

Anaconda之conda换国内源_你的大数据学长的博客-CSDN博客_conda换源

3.运行时报错

ModuleNotFoundError: No module named 'matplotlib'

下载该包 pip install matplotlib

解决 ModuleNotFoundError: No module named matplotlib.pyplot 问题_X-giraffe的博客-CSDN博客

4.运行时报错

AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'

YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘的解决方法_佐咖的博客-CSDN博客

YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_RodgeH的博客-CSDN博客

(文件中修改代码的方法,没有降低版本)

5.无法调用gpu进行训练

UserWarning: User provided device_type of ‘cuda’, but CUDA is not available. Disabling

基于pytorch的yolov5运行报错warnings.warn(‘User provided device_type of \‘cuda\‘, but CUDA is not available)_SHU_RYE的博客-CSDN博客>

解决方法:降低了版本 使用pytorch1.10+cuda10.2+cudnn8.3.3

本人RTX2060 使用pytorch最新版本+cuda11.6+cudnn8.4 pytorch给出调用为true 正确报版本

理论上cuda11.6可以用 因为RTX2060驱动为511.65 符合要求

出现问题可能如下:

1.环境变量配置 少了一条libnvvp一开始 换版本才发现

(5条消息) 【环境配置】CUDA和cuDNN安装_ztmajor的博客-CSDN博客_cudnn环境变量

2.cuda11以上版本是给30系显卡的 20系显卡只能用10版本的(我同学是这样说的 我不清楚)

你可能感兴趣的:(python,pytorch,深度学习,windows)