1.报错如下:

2.让我们进入yolov5项目,找到export.py文件,打开

1. 我的是在355行左右,可以ctrl+F检索
2. 把第3行代码注释或删除,也就是下面的代码
config.max_workspace_size = workspace * 1 << 30
3. 再添加上如下代码,就可以运行啦
config.max_workspace_size = 1 << 20
PS:如果解决了,记得给博主点赞哦


1. 我的是在355行左右,可以ctrl+F检索
2. 把第3行代码注释或删除,也就是下面的代码
config.max_workspace_size = workspace * 1 << 30
3. 再添加上如下代码,就可以运行啦
config.max_workspace_size = 1 << 20
PS:如果解决了,记得给博主点赞哦
您可能感兴趣的与本文相关的镜像
Yolo-v5
YOLO(You Only Look Once)是一种流行的物体检测和图像分割模型,由华盛顿大学的Joseph Redmon 和Ali Farhadi 开发。 YOLO 于2015 年推出,因其高速和高精度而广受欢迎
4180
3958
9241
3224