【linux】一分钟配置好yolov5

在 Ubuntu 系统下配置 YOLOv5 环境涉及几个步骤，包括安装 Python、创建虚拟环境、安装 PyTorch 和其他依赖库。以下是详细的步骤：

1. 安装 Python

Ubuntu 通常预装了 Python，但你可以通过以下命令确保 Python3 和 pip 的安装：

sudo apt update
sudo apt install python3 python3-pip

2. 创建 Python 虚拟环境

使用虚拟环境可以避免依赖冲突。安装 virtualenv 并创建一个新的环境：

pip3 install virtualenv
virtualenv yolov5env
source yolov5env/bin/activate

3. 安装 PyTorch

根据你的系统配置（是否有 GPU 支持），安装 PyTorch。可以从 PyTorch 官网 获取安装命令。对于大多数 Ubuntu 系统，如果你有 NVIDIA GPU 并安装了 CUDA，可以使用如下命令：

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

如果没有 GPU，可以安装 CPU 版本：

pip install torch torchvision torchaudio

4. 克隆 YOLOv5 仓库并安装依赖

从 GitHub 克隆 YOLOv5 仓库，并安装所需的依赖：

git clone https://github.com/ultralytics/yolov5.git
cd yolov5
pip install -r requirements.txt

5. 测试安装

运行一个简单的测试来确认 YOLOv5 是否正确安装：

python detect.py --weights yolov5s.pt --img 640 --conf 0.25 --source data/images/

这将使用预训练的模型在示例图片上进行检测。

6. 退出虚拟环境

当你完成工作后，可以通过以下命令退出虚拟环境：

deactivate

注意事项

• 确保你的 CUDA 版本与 PyTorch 版本兼容。
• 如果你的系统有多个 Python 版本，确保使用 python3 和 pip3。
• 在安装任何软件或库时，确保网络连接稳定。