YOLOV5 安装 随笔

conda：

创建3.7版本python的环境：

conda create -n yolo1.9 python=3.7

torch：

服务器cuda版本：11.1

// 官网安装合适的torch
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c conda-forge

---

requirements.txt：

注释掉yolov5中requirements.txt中的
torch>=1.7.0
torchvision>=0.8.1
避免重新覆盖安装别的版本的torch

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt

---