Install YOLO

This guide will help you set up YOLO on your machine. We recommend starting with GitHub Settings for more flexible customization. If you are planning to perform inference only or require a simple customization, you can choose to install via PyPI.

Torch Requirements

The following table summarizes the torch requirements for different operating systems and hardware configurations:

PyTorch: 1.12+

Git & GitHub

First, Clone the repository:

git clone https://github.com/WongKinYiu/YOLO.git

Alternatively, you can directly download the repository via this link.

Next, install the required packages:

# For the minimal requirements, use:
pip install -r requirements.txt
# For a full installation, use:
pip install -r requirements-dev.txt

Moreover, if you plan to utilize ONNX or TensorRT, please follow Compile to ONNX, Compile to TensorRT for more installation details.

PyPI (pip install)

Note

Due to the yolo this name already being occupied in the PyPI library, we are still determining the package name. Currently, we provide an alternative way to install via the GitHub repository. Ensure your shell has git and pip3 (or pip).

To install YOLO via GitHub:

pip install git+https://github.com/WongKinYiu/YOLO.git

Docker

To run YOLO using NVIDIA Docker, you can pull the Docker image and run it with GPU support:

docker pull henrytsui000/yolo
docker run --gpus all -it henrytsui000/yolo

Make sure you have the NVIDIA Docker toolkit installed. For more details on setting up NVIDIA Docker, refer to the NVIDIA Docker documentation.

Conda

We will publish it in the near future!