yolov7需要的torch版本
时间: 2023-07-31 20:08:27 浏览: 329
yolov5s.torchscript.pt
5星 · 资源好评率100%
YOLOv7需要的torch版本是PyTorch 1.7或更高版本。建议使用最新版本的PyTorch以确保获得最佳性能和稳定性。您可以通过以下命令检查您的PyTorch版本:
```python
import torch
print(torch.__version__)
```
如果您的PyTorch版本低于1.7,则可以通过以下命令升级:
```python
pip install torch==1.7.0 -f https://download.pytorch.org/whl/cu110/torch_stable.html
```
请注意,根据您的CUDA版本,您可能需要更改`cu110`以匹配您的CUDA版本。
阅读全文