安装 YOLOv5 依赖项 - torchvision？

问题描述

我想使用 yolov5。根据{{​​3}}，您应该安装了 Python>=3.8 和 PyTorch>=1.7，以及 YOlov5 依赖项。

Python 和 pytorch 是最新的：

pip show torch
Version: 1.7.1

python --version
Python 3.9.1

但是当我尝试安装 yolov5 依赖项时，我收到一条错误消息：

pip install -qr https://raw.githubusercontent.com/ultralytics/yolov5/master/requirements.txt
ERROR: Could not find a version that satisfies the requirement torchvision>=0.8.1
ERROR: No matching distribution found for torchvision>=0.8.1

需要更新版本的 torchvision (???)。所以我运行了更新，但是当我检查版本时它没有工作。

pip show torchvision
Version: 0.2.2.post3

-m pip install --upgrade torchvision

pip show torchvision
Version: 0.2.2.post3

安装yolov5 依赖项是否需要torchvision？我该如何前进？我使用的是 Windows 10。

谢谢！

解决方法

我遇到了类似的错误。安装新的兼容库（torch==1.7.1 & torchvision==0.8.2）对我有用。

virtualenv -p python3.8 torch17
source torch17/bin/activate
pip install cython matplotlib tqdm scipy ipython ninja yacs opencv-python ffmpeg opencv-contrib-python Pillow scikit-image scikit-learn lmfit imutils pyyaml jupyterlab==3
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

请注意您的 cuda 版本并进行相应安装。我的是10.1。只需使用 pip install torch==1.7.1 将安装最新的 cuda 版本 (11) 兼容库，这可能不是您的情况。

Windows 的虚拟环境激活命令会有所不同。我在 Linux 上