weixin_44898634的博客

原创 YoloV5环境配置问题记录

这个问题是在运行，train.py的时候产生的，同时在到训练这个步骤之前，我就发现我的程序跑得非常慢，仅仅是detect.py这个个程序就使用了10min才运行出来，当时只是在查找GPU跑得很慢的原因，怀疑没有启用GPU，于是我在我的环境中进行了测试，测试程序如下。继续找问题，这个是CSDN翻了很久，开始怀疑是pytorch版本，我的显卡是3060的，cuda是12.0的之前说是只要是在该版本以下就是支持的，我下了10.2版本的，大家测评是通用版本。----看看Pytorch参数。