conda下配置yolov5 v4.0环境

如果大家有条件的话还是建议使用科学网进行配置,避免各种问题

经历了许多次的失败,我发现还是得看官方文档,国内的这些博客,文章都没有很大的参考性,主要原因是国内源的问题,时效性很差,所以我想记录下自己刚刚配置环境的过程,保证自己在以后回看的时候还能快速的配置完成

重要的事情说3遍:一定按照官方文档来!一定按照官方文档来!一定按照官方文档来!

step1:安装anaconda

https://www.anaconda.com/products/individual

step2:查看yolov5的官方文档&明确requirements

https://github.com/ultralytics/yolov5


requirements:

python 3.8
torch>=1.7

step3:配置conda环境

conda create -n yolov5 python==3.8

conda activate yolov5

step4:安装cuda(没有GPU的话跳过,下一步也相应选择无cuda)

按照pytorch的要求按照相应版本的cuda,(不能太新也不能太旧)
https://pytorch.org/get-started/locally/

我安装的是cuda10.2
cuda的官方网址:https://developer.nvidia.com/cuda-10.2-download-archive

step5:安装pytorch


在conda env=yolov5环境下

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

step6:安装pip包

这一步有可能是大家的问题所在,可以使用清华源(亲测可行),但还是建议科学网直接下载

pip3 install -r requirements.txt     #建议使用pip3,不要使用pip(因为我pip一直失败,一用pip3就成功了)

如果失败的话,建议重头开始,包版本冲突挺麻烦的

!!!!大功告成!!!!

你可能感兴趣的:(conda下配置yolov5 v4.0环境)