"What libraries should I install to use PyTorch and YOLOv5? I am using PyCharm and Visual Studio."

[joseant7231]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

Good morning, good afternoon, or good evening. I am using Visual Studio Code and also PyCharm with the Python language. I would like to perform image recognition on a video, but I am encountering errors with Torch and YOLOv5. Could you please help me by telling me which libraries I should install first? Thank you very much for your possible response.

Additional

[UltralyticsAssistant]

👋 Hello @joseant7231, thank you for your interest in YOLOv5 🚀! This is an automated response to help you get started quickly—an Ultralytics engineer will also assist you soon.

Please visit our ⭐️ Tutorials for detailed guides—including Custom Data Training and Hyperparameter Evolution.

If this is a 🐛 Bug Report, please provide a minimum reproducible example to help us debug it.

If this is a custom training ❓ Question, please include as much information as possible, such as dataset image examples and training logs, and verify you are following our Tips for Best Training Results.

Requirements

Python>=3.8.0 with all requirements.txt installed including PyTorch>=1.8. To get started:

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

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI tests verify correct operation of YOLOv5 training, validation, inference, export and benchmarks on macOS, Windows, and Ubuntu every 24 hours and on every commit.

😊 If you encounter further issues, please provide additional details or screenshots to help us assist you!

[pderrenger]

Hi @joseant7231,

To use YOLOv5 for video recognition, install the required dependencies with:

pip install -r https://raw.githubusercontent.com/ultralytics/yolov5/master/requirements.txt

This installs PyTorch and all necessary libraries. For video processing specifically, ensure OpenCV is properly installed:

pip install opencv-python-headless

You can also clone the repository and install everything at once:

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

For detailed setup instructions, check our Quickstart guide.