在 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
。 - 在安装任何软件或库时,确保网络连接稳定。