YoloV5环境配置问题记录

最新推荐文章于 2024-10-08 12:21:22 发布

囚绿

最新推荐文章于 2024-10-08 12:21:22 发布

阅读量397

点赞数 7

文章标签： YOLO

本文链接：https://blog.csdn.net/weixin_44898634/article/details/135703211

版权

文章讲述了作者在使用PyTorch时遇到GPU运行效率低下和版本兼容性问题，通过检查驱动版本并升级到与显卡适配的版本（如30XX显卡用cu111以上），以及优化内存设置（如num_workers=0），解决了程序运行缓慢和OSError问题。

摘要由CSDN通过智能技术生成

1.AMP：checks faild/执行Detect都非常慢，感觉GPU没有启用
这个问题是在运行，train.py的时候产生的，同时在到训练这个步骤之前，我就发现我的程序跑得非常慢，仅仅是detect.py这个个程序就使用了10min才运行出来，当时只是在查找GPU跑得很慢的原因，怀疑没有启用GPU，于是我在我的环境中进行了测试，测试程序如下
import torch
----看看Cuda是否活着
Flag=torch.cuda.is_available()
print(Flag)
—看看Pytorch版本
import torch
print(torch.version)
----看看Pytorch参数
x = torch.rand(5, 3)
print(x)
以上都显示的正常

继续找问题，这个是CSDN翻了很久，开始怀疑是pytorch版本，我的显卡是3060的，cuda是12.0的之前说是只要是在该版本以下就是支持的，我下了10.2版本的，大家测评是通用版本。结果问题就在这儿。看到某博主说
“
16XX的显卡，安装cu102的版本，否则可能训练出现问题

30XX、40XX显卡，要安装cu111以上的版本，否则无法运行
”
幡然醒悟更换pytorch版本到 1.12.0+cu116

再运行，正常了，并且之前运行程序很慢的问题也同时解决了
yes!!

OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading “D:\Softw
也是查看了SCDN尝试了好多方法，以为是虚拟内存不够，该了内存
最后尝试去更改该参数值。然后解决了
网上说的地址再： yolov5\utils\datasets.py
而我改参数值的地址在：yolov5\utils\dataloaders.py
更改参数是：
num_workers=0,