安装 YOLOv5 依赖项——torchvision？

Question

我想用yolov5。根据https://pytorch.org/hub/ultralytics_yolov5/，您应该安装了 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 上。

谢谢！

Answer 1

我遇到了类似的错误。安装新的兼容库（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 上