🔥 Установка PyTorch GPU с YOLOv5

Автор: Цзин Ифу

✅ Шаг 1: Отключение зависимостей YOLOv5 PyTorch

📌 В файле requirements.txt в каталоге yolov5 отключите зависимости PyTorch:

# Base ----------------------------------------
matplotlib>=3.2.2
numpy>=1.18.5
opencv-python>=4.1.2
Pillow>=7.1.2
PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
# torch>=1.7.0 （отключить）
# torchvision>=0.8.1 （отключить）
tqdm>=4.41.0

✅ Шаг 2: Установка через официальный сайт

📌 Перейдите на официальный сайт: https://pytorch.org/get-started/locally/.

📌 Например, для Linux с версией CUDA 11.3 выберите соответствующую версию.

📌 Скопируйте команду установки и установите её в среде conda:

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

✅ Ручная загрузка (установка без подключения к сети)

Также можно вручную загрузить и установить пакеты в соответствии с вашей экспериментальной средой.

📌 Метод 1: Перейдите на страницу загрузки и найдите список пакетов torch и torchvision.

📌 Метод 2: Перейдите на сайт с указанием версии CUDA, например, https://download.pytorch.org/whl/cu113 для CUDA 11.3.

https://download.pytorch.org/whl/cu**

📌 Выберите пакеты torch и torchvision, соответствующие вашей экспериментальной среде, включая операционную систему, тип CUDA, версию Python и т. д.

📌 Установите загруженные пакеты PyTorch с помощью pip:

pip install ./torch-***.whl # сначала установить
pip install ./torchvision-***.whl # затем установить

❗ Обратите внимание: существует порядок установки, поскольку torch является родительским пакетом для torchvision. Если вы сначала установите torchvision, он автоматически загрузит пакет torch. Поэтому сначала установите torch, а затем torchvision.

❤️ Версия PyTorch

📌 Официальный сайт: https://github.com/pytorch/pytorch/wiki/PyTorch-Versions.

💡 Просмотр версий PyTorch

import torch

print(torch.__version__)  # просмотр версии PyTorch
print(torch.version.cuda)  # просмотр версии CUDA
print(torch.backends.cudnn.version())  # просмотр версии cuDNN