Skip to main content

Yolov5 安装与环境配置

官方参阅链接：ultralytics/yolov5: YOLOv5 🚀 in PyTorch

yolov5基于pytorch，因此我们需要先安装pytorch

安装pytorch​

官方参阅链接：Previous PyTorch Versions | PyTorch

  • 找到官方最新版本，选择自己硬件适合的版本，如果python是使用的conda则使用 conda install，原生环境使用pip

注意：ubuntu一般有两个python（python2 python3），pip默认链接到的是python2.7，我们需要灵活使用pip3来安装

注意：如果提示没有pip3，则先sudo apt-get install python3-pip，然后升级到最新版本 pip3 install --upgrade pip，一定要升级到最新版本，不然下面步骤会有很多错误，切记

  • 版本主要分为CUDA版本和CPU版本，如果你有NVIDIA的显卡，则使用CUDA版本，否则使用CPU版本
# CUDA 11.1
pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 10.2
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
  • 安装pytorch，这里我安装的是CPU版本，注意pip3，如果是conda则使用conda替换pip3
pip3 install torch==1.10.1+cpu torchvision==0.11.2+cpu torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
  • 没有报错则认为完成，进行下一步

安装yolov5​

注意前置条件：Python>=3.6.0 is required with all requirements.txt installed including PyTorch>=1.7:

  • 使用以下命令安装yolov5，这里我也改成了pip3，根据实际环境
git clone https://github.com/ultralytics/yolov5
cd yolov5
pip3 install -r requirements.txt
  • 同样，没有报错即可

至此安装完毕，有报错要看是什么错，对应解决，再重新来。

Last updated on