0 安装对应得cuda版本,点击下面链接进行下载
1 创建虚拟环境
conda create -n yolov5 python=3.9
conda:指调用conda包
create:指创建的意思
-n:是指环境的名称
yolov5:是环境的名称
python=3.9:是创建的屋子,是python3.9版本
2 激活虚拟环境
conda activate yolov5
3 安装镜像源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
4 安装pytorch
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
5 安装requirements依赖
cd 到 yolov5 目录下
pip install -r requirements