yolov5的环境配置

达拉达拉开

于 2022-04-20 15:59:41 发布

阅读量148

点赞数

CC 4.0 BY-SA版权

文章标签：算法 python

本文链接：https://blog.youkuaiyun.com/weixin_44633205/article/details/123756502

本文记录了一位新手成功配置Yolov5运行环境的过程，包括更换PyTorch源和使用清华源，以及通过修改batchsize解决CUDA内存不足的问题。此外，还分享了当pip安装库失败时的备用方案——手动安装并复制到site-packages。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

前言

新手小白终于配置好了yolov5的运行环境，喜大普奔！写一些自己想要记忆的东西。

一、换源

pytorch -c+新的源（比如清华源）

二、安装库

pip install -r requirements.txt

如果安装不成功，可以直接下载安装包，然后复制到\Lib\site-packages，亲测有用（pycocotools）


untimeError: CUDA out of memory. Tried to allocate 2.27 GiB (GPU 0; 11.00 GiB total capacity; 517.31 MiB already allocated; 8.32 GiB free; 866.00 MiB reserved in total by PyTorch) If reserved memory is >> allocated memory try ...
解决方法：batchsize==1


---