安装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？我该如何前进？我在Windows10上

谢谢

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）兼容的lib，这可能不是您的情况

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