yolov8环境配置

最新推荐文章于 2024-07-25 23:03:08 发布

想为祖国献石油

最新推荐文章于 2024-07-25 23:03:08 发布

阅读量478

点赞数 5

文章标签： YOLO

本文链接：https://blog.csdn.net/weixin_47208848/article/details/136758167

版权

首先ananconda创建虚拟环境，python版本最好大于3.8
再配gpu版本的torch。

pip install torch==2.1.0+cu118 torchvision==0.16.0+cu118 torchaudio==2.1.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/ some-package --trusted-host mirrors.aliyun.com

安装额外的包

 pip install timm==0.9.8 thop efficientnet_pytorch==0.7.1 einops grad-cam==1.4.8 dill==0.3.6 albumentations==1.3.1 pytorch_wavelets==1.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

以下主要是使用dyhead必定需要安装的包

pip install -U openmim
mim install mmengine -i https://pypi.tuna.tsinghua.edu.cn/simple
mim install "mmcv>=2.0.0" -i https://pypi.tuna.tsinghua.edu.cn/simple