You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
132 lines
5.2 KiB
132 lines
5.2 KiB
--- |
|
comments: true |
|
description: Comprehensive guide to setting up and using Ultralytics YOLO models in a Conda environment. Learn how to install the package, manage dependencies, and get started with object detection projects. |
|
keywords: Ultralytics, YOLO, Conda, environment setup, object detection, package installation, deep learning, machine learning, guide |
|
--- |
|
|
|
# Conda Quickstart Guide for Ultralytics |
|
|
|
<p align="center"> |
|
<img width="800" src="https://user-images.githubusercontent.com/26833433/266324397-32119e21-8c86-43e5-a00e-79827d303d10.png" alt="Ultralytics Conda Package Visual"> |
|
</p> |
|
|
|
This guide provides a comprehensive introduction to setting up a Conda environment for your Ultralytics projects. Conda is an open-source package and environment management system that offers an excellent alternative to pip for installing packages and dependencies. Its isolated environments make it particularly well-suited for data science and machine learning endeavors. For more details, visit the Ultralytics Conda package on [Anaconda](https://anaconda.org/conda-forge/ultralytics) and check out the Ultralytics feedstock repository for package updates on [GitHub](https://github.com/conda-forge/ultralytics-feedstock/). |
|
|
|
[](https://anaconda.org/conda-forge/ultralytics) [](https://anaconda.org/conda-forge/ultralytics) [](https://anaconda.org/conda-forge/ultralytics) [](https://anaconda.org/conda-forge/ultralytics) |
|
|
|
## What You Will Learn |
|
|
|
- Setting up a Conda environment |
|
- Installing Ultralytics via Conda |
|
- Initializing Ultralytics in your environment |
|
- Using Ultralytics Docker images with Conda |
|
|
|
--- |
|
|
|
## Prerequisites |
|
|
|
- You should have Anaconda or Miniconda installed on your system. If not, download and install it from [Anaconda](https://www.anaconda.com/) or [Miniconda](https://docs.conda.io/projects/miniconda/en/latest/). |
|
|
|
--- |
|
|
|
## Setting up a Conda Environment |
|
|
|
First, let's create a new Conda environment. Open your terminal and run the following command: |
|
|
|
```bash |
|
conda create --name ultralytics-env python=3.8 -y |
|
``` |
|
|
|
Activate the new environment: |
|
|
|
```bash |
|
conda activate ultralytics-env |
|
``` |
|
|
|
--- |
|
|
|
## Installing Ultralytics |
|
|
|
You can install the Ultralytics package from the conda-forge channel. Execute the following command: |
|
|
|
```bash |
|
conda install -c conda-forge ultralytics |
|
``` |
|
|
|
### Note on CUDA Environment |
|
|
|
If you're working in a CUDA-enabled environment, it's a good practice to install `ultralytics`, `pytorch`, and `pytorch-cuda` together to resolve any conflicts: |
|
|
|
```bash |
|
conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda=11.8 ultralytics |
|
``` |
|
|
|
--- |
|
|
|
## Using Ultralytics |
|
|
|
With Ultralytics installed, you can now start using its robust features for object detection, instance segmentation, and more. For example, to predict an image, you can run: |
|
|
|
```python |
|
from ultralytics import YOLO |
|
|
|
model = YOLO('yolov8n.pt') # initialize model |
|
results = model('path/to/image.jpg') # perform inference |
|
results[0].show() # display results for the first image |
|
``` |
|
|
|
--- |
|
|
|
## Ultralytics Conda Docker Image |
|
|
|
If you prefer using Docker, Ultralytics offers Docker images with a Conda environment included. You can pull these images from [DockerHub](https://hub.docker.com/r/ultralytics/ultralytics). |
|
|
|
Pull the latest Ultralytics image: |
|
|
|
```bash |
|
# Set image name as a variable |
|
t=ultralytics/ultralytics:latest-conda |
|
|
|
# Pull the latest Ultralytics image from Docker Hub |
|
sudo docker pull $t |
|
``` |
|
|
|
Run the image: |
|
|
|
```bash |
|
# Run the Ultralytics image in a container with GPU support |
|
sudo docker run -it --ipc=host --gpus all $t # all GPUs |
|
sudo docker run -it --ipc=host --gpus '"device=2,3"' $t # specify GPUs |
|
``` |
|
|
|
--- |
|
|
|
Certainly, you can include the following section in your Conda guide to inform users about speeding up installation using `libmamba`: |
|
|
|
--- |
|
|
|
## Speeding Up Installation with Libmamba |
|
|
|
If you're looking to [speed up the package installation](https://www.anaconda.com/blog/a-faster-conda-for-a-growing-community) process in Conda, you can opt to use `libmamba`, a fast, cross-platform, and dependency-aware package manager that serves as an alternative solver to Conda's default. |
|
|
|
### How to Enable Libmamba |
|
|
|
To enable `libmamba` as the solver for Conda, you can perform the following steps: |
|
|
|
1. First, install the `conda-libmamba-solver` package. This can be skipped if your Conda version is 4.11 or above, as `libmamba` is included by default. |
|
|
|
```bash |
|
conda install conda-libmamba-solver |
|
``` |
|
|
|
2. Next, configure Conda to use `libmamba` as the solver: |
|
|
|
```bash |
|
conda config --set solver libmamba |
|
``` |
|
|
|
And that's it! Your Conda installation will now use `libmamba` as the solver, which should result in a faster package installation process. |
|
|
|
--- |
|
|
|
Congratulations! You have successfully set up a Conda environment, installed the Ultralytics package, and are now ready to explore its rich functionalities. Feel free to dive deeper into the [Ultralytics documentation](../index.md) for more advanced tutorials and examples.
|
|
|