如果大家有条件的话还是建议使用科学网进行配置,避免各种问题
经历了许多次的失败,我发现还是得看官方文档,国内的这些博客,文章都没有很大的参考性,主要原因是国内源的问题,时效性很差,所以我想记录下自己刚刚配置环境的过程,保证自己在以后回看的时候还能快速的配置完成
重要的事情说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就成功了)
如果失败的话,建议重头开始,包版本冲突挺麻烦的