安装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