-
Установка NVIDIA CUDA Toolkit 12.8 (глобально) Для Linux:
wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_570.86.10_linux.run sudo sh cuda_12.8.0_570.86.10_linux.run
-
Создание и активация виртуального окружения Conda
- Создайте окружение с именем
gs_envи Python 3.11:conda create -n gs_env python=3.11 -y
- Активируйте созданное окружение:
conda activate gs_env
- Создайте окружение с именем
-
Установка PyTorch для CUDA 12.8 Установите совместимые версии
torchиtorchvision:pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128
-
Установка фреймворка Ultralytics (YOLO) Установите пакет
ultralyticsдля работы с YOLO:pip install ultralytics
- Для глобальной установки рекомендуется использовать CUDA Toolkit версии 12.8, так как это стабильная и проверенная версия (см. пункт 1 инструкции).
- При создании виртуального окружения рекомендуется выбрать Python версии 3.11, как одну из наиболее стабильных и совместимых версий (см. пункт 2 инструкции).
- Установку PyTorch и Ultralytics (YOLO) (пункты 3 и 4) необходимо выполнять исключительно с помощью
pip. Использование менеджера пакетовcondaдля этих библиотек может привести к ошибкам совместимости между YOLO и PyTorch.