Как установить Ultralytics YOLO с помощью Conda#
Это руководство поможет тебе настроить среду Conda для проектов Ultralytics. Conda — это система управления пакетами и средами с открытым исходным кодом, которая служит отличной альтернативой pip при установке пакетов и зависимостей. Изолированные среды делают её особенно подходящей для задач в области анализа данных и машинного обучения. Подробнее смотри на странице пакета Ultralytics в Anaconda и в репозитории feedstock Ultralytics с обновлениями пакета на GitHub.
В этом руководстве рассказывается, как создать среду, установить Ultralytics, запустить инференс, использовать образ Conda для Docker и ускорить установку с помощью libmamba.
Предварительные требования#
В твоей системе должны быть установлены Anaconda или Miniconda. Если их нет, скачай и установи один из них с сайта Anaconda или Miniconda.
Настройка среды Conda#
Сначала создай новую среду Conda. Открой терминал и выполни следующую команду:
conda create --name ultralytics-env python=3.11 -yАктивируй новую среду:
conda activate ultralytics-envУстановка Ultralytics#
Ты можешь установить пакет Ultralytics из канала conda-forge. Выполни следующую команду:
conda install -c conda-forge ultralyticsЕсли ты работаешь в среде с поддержкой CUDA, рекомендуется установить ultralytics, pytorch и pytorch-cuda вместе, чтобы менеджер пакетов Conda мог разрешить возможные конфликты:
conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda=12.1 ultralyticsИспользование Ultralytics#
После установки Ultralytics ты можешь использовать его широкие возможности для обнаружения объектов, сегментации экземпляров и других задач. Например, чтобы выполнить предсказание для изображения, запусти:
from ultralytics import YOLO
model = YOLO("yolo26n.pt") # initialize model
results = model("path/to/image.jpg") # perform inference
results[0].show() # display results for the first imageОбраз Ultralytics Conda для Docker#
Если ты предпочитаешь использовать Docker, Ultralytics предлагает образы Docker со встроенной средой Conda. Ты можешь загрузить эти образы с DockerHub.
Загрузи последний образ Ultralytics:
# 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 Ultralytics image in a container with GPU support
sudo docker run -it --ipc=host --device nvidia.com/gpu=all $t # all GPUs
sudo docker run -it --ipc=host --device nvidia.com/gpu=2 --device nvidia.com/gpu=3 $t # specify GPUsВ Linux запросы устройств CDI требуют Docker >= 28.2.0 и NVIDIA Container Toolkit >= 1.18. Устаревший флаг --gpus all может привести к потере доступа к GPU после перезагрузки фоновых служб хоста, поэтому обнови старые хосты Linux и используй вместо него --device. Подробнее смотри в кратком руководстве по Docker.
Ускорение установки с помощью Libmamba#
libmamba — это быстрый кроссплатформенный решатель с учётом зависимостей, который заменяет классический решатель Conda. В Conda 23.10 и более поздних версиях libmamba уже используется как решатель по умолчанию, поэтому большинство установок сразу выполняется быстрее.
Если у тебя старая версия Conda, ты можешь вручную включить libmamba:
-
Сначала установи пакет
conda-libmamba-solver:conda install conda-libmamba-solver -
Затем настрой Conda на использование
libmambaв качестве решателя:conda config --set solver libmamba
Ты успешно настроил среду Conda и установил пакет Ultralytics. Теперь ты готов изучить его возможности. Дополнительные руководства и примеры смотри в документации Ultralytics.
Часто задаваемые вопросы#
Настроить среду Conda для проектов Ultralytics довольно просто, и это обеспечивает удобное управление пакетами. Сначала создай новую среду Conda с помощью следующей команды:
conda create --name ultralytics-env python=3.11 -yЗатем активируй новую среду командой:
conda activate ultralytics-envНаконец, установи Ultralytics из канала conda-forge:
conda install -c conda-forge ultralyticsConda — это надёжная система управления пакетами и средами, которая имеет ряд преимуществ перед pip. Она эффективно управляет зависимостями и обеспечивает совместимость всех необходимых библиотек. Изолированные среды Conda предотвращают конфликты между пакетами, что особенно важно в проектах по анализу данных и машинному обучению. Кроме того, Conda поддерживает распространение бинарных пакетов, ускоряя процесс установки.
Да, ты можешь повысить производительность, используя среду с поддержкой CUDA. Установи
ultralytics,pytorchиpytorch-cudaвместе, чтобы избежать конфликтов:conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda=12.1 ultralyticsТакая настройка обеспечивает ускорение на GPU, необходимое для ресурсоёмких задач, таких как обучение моделей глубокого обучения и инференс. Подробнее смотри в руководстве по установке Ultralytics.
Использование образов Ultralytics Docker обеспечивает согласованную и воспроизводимую среду, устраняя проблемы в духе «у меня на компьютере работает». Эти образы включают предварительно настроенную среду Conda, что упрощает процесс настройки. Ты можешь загрузить и запустить последний образ Ultralytics Docker с помощью следующих команд:
sudo docker pull ultralytics/ultralytics:latest-conda sudo docker run -it --ipc=host --device nvidia.com/gpu=all ultralytics/ultralytics:latest-conda # all GPUs sudo docker run -it --ipc=host --device nvidia.com/gpu=2 --device nvidia.com/gpu=3 ultralytics/ultralytics:latest-conda # specify GPUsЭтот подход идеально подходит для развёртывания приложений в рабочей среде или запуска сложных рабочих процессов без ручной настройки. Подробнее смотри в разделе Образ Ultralytics Conda для Docker.
В Conda 23.10 и более поздних версиях быстрый решатель
libmambaуже используется по умолчанию. В старых версиях Conda ты можешь включить его вручную, сначала установив пакетconda-libmamba-solver:conda install conda-libmamba-solverЗатем настрой Conda на использование
libmambaв качестве решателя:conda config --set solver libmambaТакая настройка обеспечивает более быстрое и эффективное управление пакетами. Дополнительные советы по оптимизации среды смотри в материале об установке libmamba.