yolov5安装过程记录
最最最最重要的事情就是版本,不仅torch和torchvision要对应,跟Python的版本也要对应!下图是对照表,大家根据需求和对照表来选择版本
安装yolov5方法链接【手把手带你实战YOLOv5-入门篇】YOLOv5 环境安装_哔哩哔哩_bilibili
先装cuda 后装anaconda 再装pytorch 再装包requirements.txt
安装anaconda方法和以上链接视频中安装miniconda方法一样
安装所需文件
Anaconda3-2021.05-Windows-x86_64.exe
cuda_11.6.0_511.23_windows.exe
cudnn-windows-x86_64-8.4.0.27_cuda11.6-archive.zip
pytorch1.12.0安装命令
# CUDA 11.6
pip install torch==1.12.0+cu116 torchvision==0.13.0+cu116 torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu116
一 、GPU算力查看链接:https://developer.nvidia.cn/zh-cn/cuda-gpus 如下图
cuda下载链接
CUDA Toolkit Archive | NVIDIA Developer
cudnn下载链接
安装cuda方法
【CUDA安装/多CUDA兼容】Windows深度学习环境配置_哔哩哔哩_bilibili
二、anaconda镜像链接