Skip to content

akovin/metal_sheet_estimation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Инструкция по настройке среды для запуска YOLO

  1. Установка 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
  2. Создание и активация виртуального окружения Conda

    • Создайте окружение с именем gs_env и Python 3.11:
      conda create -n gs_env python=3.11 -y
    • Активируйте созданное окружение:
      conda activate gs_env
  3. Установка PyTorch для CUDA 12.8 Установите совместимые версии torch и torchvision:

    pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128
  4. Установка фреймворка Ultralytics (YOLO) Установите пакет ultralytics для работы с YOLO:

    pip install ultralytics

Важные замечания:

  1. Для глобальной установки рекомендуется использовать CUDA Toolkit версии 12.8, так как это стабильная и проверенная версия (см. пункт 1 инструкции).
  2. При создании виртуального окружения рекомендуется выбрать Python версии 3.11, как одну из наиболее стабильных и совместимых версий (см. пункт 2 инструкции).
  3. Установку PyTorch и Ultralytics (YOLO) (пункты 3 и 4) необходимо выполнять исключительно с помощью pip. Использование менеджера пакетов conda для этих библиотек может привести к ошибкам совместимости между YOLO и PyTorch.

About

Автоматический расчет геометрических размеров листов металла на основе компьютерного зрения

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors