yolov8环境安装
#创建环境
conda create -n ultralytics python=3.9
#配置源
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
#安装pytorch,测试过了,这个版本稳定
pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu118
#安装ultralytics
cmd进入ultralytics-8.1.0, 并激活:conda activate [ultralytics,环境名]
然后:pip install -e .
备注:pytorch版本选择
终端输入:nvida-smi 查看CUDA版本,这个显示的是支持的最高的。 安装比它低就行
YOLOv5:
第一步:
conda create -n yolov5 python=3.8.18
第二步:pip换版本
python.exe -m pip install pip==23.3
第三步:
pip install wheel==0.41.2 setuptools==68.0.0
第四步:
pip install torch==1.8.2 torchvision==0.9.2 torchaudio==0.8.2 --extra-index-url https://download.pytorch.org/whl/lts/1.8/cu111
第五步:(要调整好requirements中各种包的版本)
pip install -r requirements.txt