Win10+pytorch настраивает набор данных для обучения сети yolov5
1. Программное обеспечение
Win10+Anaconda3+pytorch2.0.0+python3.9.16+opencv_python4.7
1.1 Установите Анаконду3
Установочный пакет исходного кода Tsinghua: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Выбрано
для этой статьи: Anaconda3-2022.10-Linux-x86_64.sh
1.1.1 Установка
Маггловская установка: «Следующий шаг», но не выбирайте «Добавить переменные среды» на последнем шаге
***: Не задавайте китайский путь для папки.
1.1.2 Конфигурация среды
Добавьте соответствующие пути установки в «путь» пользовательских переменных и системных переменных соответственно.
Добавьте источник загрузки:
откройте терминал Anaconda Prompot (Anaconda) и введите:
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
Посмотреть адрес канала:
conda config --show channels
***: Это также можно реализовать напрямую, изменив конфигурацию пользователя, то есть добавив указанный выше источник в файл C:\Users\xxx.condarc.
1.1.3 Проверка установки
Откройте терминал cmd и введите:
conda --version //conda 22.9.0
conda info
1.2 Установка виртуальной среды Pytorch
В этой таблице приведены различные версии системы, методы установки, версии Python и версии CUDA, поддерживаемые pytorch (то есть официальным драйвером инструмента ускорения графического процессора NVIDIA): https://pytorch.org/get-started/locally/Native
Configuration: CUDA11. 8, python3.9
открывает терминал Anaconda Prompt.
conda create -n pytorch python=3.9
activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install -c menpo opencv
python
import torch
x = torch.rand(5, 3)
print(x)
#tensor([[0.3380, 0.3845, 0.3217],
# [0.8337, 0.9050, 0.2650],
# [0.2979, 0.7141, 0.9069],
# [0.1449, 0.1132, 0.1375],
# [0.4675, 0.3947, 0.1426]])
import torch
torch.cuda.is_available()
***若下载速度慢，更新conda国内源
清华源：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
1.3 Установите LabelImg
Загрузка исходного кода: https://gitcode.net/mirrors/tzutalin/labelimg?utm_source=csdn_github_accelerator
Откройте терминал Anaconda Prompt.
conda create -n labelimg
activate labelimg
conda install lxml
conda install PyQt5
cd E:xxx #//安装包路径
Pyrcc5 -o resources.py resources.qrc #会在当前目录下生成一个resource.py