1.可用环境

torch==1.7.1+cu110
torchvision==0.8.2+cu110

2.错误环境遇到的问题

（1）情况一

torch==1.7.0+cu110
torchvision==0.8.0

训练yolov5时，训练就会开始报错：

return torch.ops.torchvision.nms(boxes, scores, iou_threshold)
RuntimeError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. 'torchvision::nms' is only available for these backends: [CPU, BackendSelect, Named, AutogradOther, AutogradCPU, AutogradCUDA, AutogradXLA, Tracer, Autocast, Batched, VmapMode].

torchvision版本不对。所以升级了torchvision的版本。进入到情况二。

（2）情况二

torch==1.7.0+cu110
torchvision==0.8.1

训练第一个epoch时，不出错，但是测试第一个epoch的时候，会报错：

【python】yolov5的torch与torchvision环境问题相关推荐

Pytorch和Torch和torchvision的安装
超详细的Pytorch和torch和torchvision的安装步骤~~~~~ Pytorch安装步骤包地址:https://download.pytorch.org/whl/torch_stabl ...
Torch和torchvision的安装
叮~ 快收藏torch和torchvision的详细安装步骤~~~~~ 要安装torch和torchvision,首先要确定你电脑安装的python的版本,而且还要知道torch和torchvisio ...
YOLOv5错误提示（ torchvision和torch版本不兼容）
错误提示是 torchvision和torch的版本不兼容 File "D:\Programs\conda411\envs\envZhao\lib\site-packages\torchvi ...
Python: 如何安装 torch 对应的 torchvision 版本？
Python: 如何安装 torch 对应的 torchvision 版本? 参考链接:https://pytorch.org/get-started/locally/ 选择合适的版本: 运行命令: ...
树莓派python中安装numpy，pandas，torch，torchvision失败，ERROR: Could not find a version that satisfies the req
树莓派python中安装numpy,pandas,torch,torchvision问题安装失败现象 python通过pip安装torch错误ERROR: Could not find a vers ...
torch 和torchvision对应关系并附下载路径
https://blog.csdn.net/AugustMe/article/details/109028989 torch对应的torchvision版本版权安装torch和torchvisio ...
NVIDIA Jetson Xavier NX安装torch和torchvision
文章目录: 1 安装torch 2 安装torchvision 2.1 方法一:直接源码安装torchvision 2.2 方法二:使用pip3安装torchvision 1 安装torch 1.教程 ...
关于安装torch、torchvision包的问题
关于安装torch.torchvision包的问题问题描述: 1. 无法下载pytorch包已尝试的方法如下: conda安装pytorch下载速度过慢(Windows下)–使用清华镜像 pyto ...
Ubuntu18.04.1系统安装mmdetection(含torch、torchvision、mmcv-full)
临近春节,实验室机房装修,导师放假.回家后便于学习,在自己的笔记本上也装了一下mmdetection,分享一下个人的安装过程,方便以后安装少走弯路以及供大家参考交流. 我的电脑装的是双系统,win10 ...

【python】yolov5的torch与torchvision环境问题

1.可用环境

2.错误环境遇到的问题

（1）情况一

（2）情况二

【python】yolov5的torch与torchvision环境问题相关推荐

最新文章

热门文章